Ubuntu20.04安装cuda10.1的步骤(图文教程)

下面是Ubuntu20.04安装cuda10.1的步骤详细攻略：

1. 准备工作

• 操作系统：Ubuntu 20.04
• 显卡驱动：建议使用官方推荐驱动或更高版本
• CUDA版本：CUDA 10.1

2. 下载并安装CUDA Toolkit

首先从Nvidia官网上下载CUDA Toolkit 10.1，可以通过WGET命令或浏览器下载，这里以WGET命令为例：

$ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
$ sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
$ wget https://developer.download.nvidia.com/compute/cuda/10.1/secure/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
$ sudo sh cuda_10.1.243_418.87.00_linux.run

安装过程中会提示一些选项，一般默认即可。

3. 配置系统环境

安装完成后，需要配置CUDA相关的环境变量，以便其他程序可以调用CUDA相关库文件。

$ echo 'export PATH=/usr/local/cuda-10.1/bin:$PATH' >> ~/.bashrc
$ echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
$ source ~/.bashrc

4. 验证安装成功

安装完成后，我们需要验证一下是否安装成功。可以使用以下命令查看CUDA版本：

$ nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Sun_Jul_28_19:07:16_PDT_2019
Cuda compilation tools, release 10.1, V10.1.243

如果能显示出CUDA的版本信息，则说明安装成功了。

示例说明：

示例一：安装TensorFlow-gpu

安装了CUDA后，我们可以使用pip命令安装TensorFlow-gpu

$ pip install tensorflow-gpu

此时，如果CUDA安装成功，TensorFlow-gpu会默认调用正确的CUDA库文件，加速运算。

示例二：编译运行CUDA程序

安装成功后，我们可以尝试编译运行CUDA程序。例如，可以使用以下代码编译运行一个简单的加法程序，它会使用CUDA进行并行计算：

#include <stdio.h>

__global__ void add(int a, int b, int *c) {
    *c = a + b;
}

int main(void) {
    int c;
    int *dev_c;
    cudaMalloc((void **) &dev_c, sizeof(int));

    add<<<1,1>>>(2, 7, dev_c);

    cudaMemcpy(&c, dev_c, sizeof(int), cudaMemcpyDeviceToHost);

    printf("2 + 7 = %d\n", c);

    cudaFree(dev_c);
    return 0;
}

编译运行：

$ nvcc -o add add.cu
$ ./add

若能成功输出 "2 + 7 = 9"，则说明CUDA编译运行成功。

以上就是在Ubuntu20.04上安装CUDA 10.1的详细步骤说明。