ubuntu 20.04安装cuda toolkit 11.6以Thinkpad-Quadro-P620为例

—————————————————— 前置条件 —————————————————————————

1.参考上篇博客，安装nvidia驱动

2.验证nvidia驱动安装是否成功：

nvidia-smi

输出类似如下的内容：

Mon Aug  1 13:30:11 2022       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 515.48.07    Driver Version: 515.48.07    CUDA Version: 11.7     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  Quadro P620         Off  | 00000000:01:00.0 Off |                  N/A |
| N/A   41C    P8    N/A /  N/A |      9MiB /  4096MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1376      G   /usr/lib/xorg/Xorg                  4MiB |
|    0   N/A  N/A      1970      G   /usr/lib/xorg/Xorg                  4MiB |
+-----------------------------------------------------------------------------+

—————————————————————————— 正文 ————————————————————————————

1. 确定安装目的：推理和训练AI模型，需要用到pytorch-gpu，从官网得知，pytorch最新支持cuda-toolkit-11.6，因此依赖链如下：
   pytorch-gpu(1.12.0) -> cuda-toolkit(11.6.2) -> nvidia-driver-515(最高支持cuda-toolkit-11.7)

2. 打开nvidia官网下载cuda-toolkit-11.6.2的runfile，网址如下：
   https://developer.nvidia.com/cuda-11-6-2-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local
   下载及安装命令如下：

wget https://developer.download.nvidia.com/compute/cuda/11.6.2/local_installers/cuda_11.6.2_510.47.03_linux.run
sudo sh cuda_11.6.2_510.47.03_linux.run

安装时请不要选择driver，因为之前已经安装了显卡驱动，仅选择toolkit-11.6就行。
安装完成后，按照提示，在~/.bashrc文件中添加环境变量：

# CUDA toolkit setup
export PATH=$PATH:/usr/local/cuda-11.6/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.6/lib64

然后保存退出后，执行以下命令使得环境变量生效：

source .bashrc

3.测试cuda是否安装成功：

nvcc --version

输出类似如下：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_Mar__8_18:18:20_PST_2022
Cuda compilation tools, release 11.6, V11.6.124
Build cuda_11.6.r11.6/compiler.31057947_0

表明cuda-toolkit安装成功。