文章目录

• 1.前置知识
• 2.查看显卡驱动版本号
• 3.查看显卡驱动版本号和CUDA版本对应关系
• 4.查看经典的CUDA版本号
• 5.安装CUDA
• • 5.1.下载CUDA安装包
  • 5.2.执行CUDA安装
  • 5.3.配置环境变量
  • 5.4.CUDA多版本管理

1.前置知识

如果Ubuntu系统还没有安装显卡驱动，参考这篇文章：Ubuntu20.04LTS安装RTX-3060显卡驱动

2.查看显卡驱动版本号

当显卡驱动安装完成后，需要使用 nvidia-smi 命令查看英伟达显卡驱动版本。

nvidia-smi

如上图所示，英伟达驱动版本为520.61.05，CUDA最高支持的版本为11.8。

3.查看显卡驱动版本号和CUDA版本对应关系

点击该链接：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html，查看英伟达显卡驱动版本和CUDA版本的对应关系。

下图为CUDA工具包和CUDA小版本兼容性所需的最低驱动程序版本：

由于我工作站（Ubuntu 20.04 LTS）的英伟达驱动版本为520.61.05，从上图可以看出，我最高可以安装的CUDA版本为11.8.x。（注：CUDA 12.0.x和CUDA 12.1.x都要求英伟达驱动版本大于等于525.60.13，因此我的520.61.05不符合，所以我最高只能安装CUDA 11.8.x的版本）。

4.查看经典的CUDA版本号

由于我们最终是要安装pytorch，因此选取合适的CUDA进行安装是是否有必要的。通过该链接，可以查看 CUDA与PyTorch版本对应关系。

从链接可以看出，10.2、11.1、11.3这三个版本比较经典，能够支持多数pytorch版本，因此可以选择这三个版本的CUDA进行安装。

5.安装CUDA

5.1.下载CUDA安装包

CUDA官方下载链接：https://developer.nvidia.com/cuda-toolkit-archive

进入CUDA官方的下载链接后，查找自己需要下载的版本（以CUDA 11.3.1为例）：

点击自己需要下载的版本，一次选择操作系统、系统架构、系统版本和安装方式，在这里推荐使用 runfile(local) 的安装方式。

完成上述操作后，网页下方弹出安装的命令，如下图所示：

5.2.执行CUDA安装

在Ubuntu 20.04 LTS系统的命令行中，按照以下命令进行安装。

# 第一步：使用wget命令下载安装包
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run`

# 第二步：执行安装脚本
sudo sh cuda_11.3.1_465.19.01_linux.run

执行上述命令后，等待1分钟左右，系统会弹出安装的协议，问你 Do you accept the above EULA? 你需要在后面的光标处，填写 accept，然后敲回车。

然后系统询问安装的内容，注意！！！ 一定要把Driver驱动这个给去掉（按空格键可以将 X 去掉），如果[ ]内是X 说明是要安装的；如果[ ] 是空，说明不安装。选择完成后，然后移动至Install 处，敲击回车。

5.3.配置环境变量

使用 vim ~/.bashrc 命令进行编辑，在文件末尾添加下列代码：

# cuda
export LD_LIBRARY_PATH=/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin

然后执行 source ~/.bashrc 刷新文件使其生效。

5.4.CUDA多版本管理

从图中可以看出，系统安装了11.1、11.3和11.6版本。

由于环境变量的地址为 /usr/local/cuda，且我们可以从图中发现目录下存在一个软链接，即：/usr/local/cuda 指向了 /usr/local/cuda-11.6 链接，说明此时尽管安装了CUDA 11.1和 CUDA 11.3版本，但系统默认的环境版本为11.6。如何进行多版本的切换呢，比如想把CUDA版本切换成11.3版本（但是要保留CUDA 11.1和11.6版本），我们只需要修改软链接即可，将CUDA 11.3的软链接链接到cuda目录下，代码如下：

sudo rm -rf cuda  # 删除原有的软链接
sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda  # 将cuda-11.3链接到cuda下

软链接重新生成后，使用 nvcc -V 命令可以查看当前的CUDA版本，如下所示：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Mar_21_19:15:46_PDT_2021
Cuda compilation tools, release 11.3, V11.3.58
Build cuda_11.3.r11.3/compiler.29745058_0