前言

當我們復現代碼時，總會需要不同版本的 CUDA 。此時，我們就需要安裝多個版本的 CUDA ，并實現靈活切換。

本文將在已有 CUDA11.3 的 Ubuntu20.04 下，安裝 CUDA11.8 。

一、安裝 CUDA

1.找到所需版本對應命令

CUDA官方下載地址

1.1 找到對應版本

1.2 選擇對應的操作系統
Linux -> x86_64 -> Ubuntu -> 20.04 -> runfile（local）

可以看到有兩條命令，依次對應下載和安裝。

2.下載 .run 文件

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run

3.安裝 CUDA

(1) 運行 .run 文件進行安裝

sudo sh cuda_11.8.0_520.61.05_linux.run

(2) 方向下鍵↓，選中 “Continue” ，回車 “Enter” 確認

(3) 輸入 accept，然后回車

(4) 使用方向下鍵↓，移動到 Driver 上，回車勾選掉 Driver
由于安裝的多版本 CUDA ，安裝第一個 CUDA 的時候已經安裝過驅動，所以不需要再重復安裝。

(5) 使用方向下鍵↓，移動到 Install 上，回車開始安裝

(6) 之后一路默認，一直敲回車即可

出現如下圖的提示即是安裝成功

4.配置環境變量

4.1 寫入環境變量

(1) 打開配置文件

sudo gedit ~/.bashrc

(2) 寫入所需的環境變量

# CUDA enviroment variable
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

(3) 更新環境變量

source ~/.bashrc

4.2 軟連接

(1) 進入 /usr/local 目錄下

cd /usr/local

(2) 刪除原有的軟連接

sudo rm -rf cuda

(3) 新建軟連接，將其連接到 CUDA11.8

sudo ln -s /usr/local/cuda-11.8 /usr/local/cuda

5.驗證安裝

nvcc -V

可以看到 CUDA 版本已經更新為 11.8

二、安裝 cudnn

1.下載 cudnn

CUDA官方網址
選擇適合自己 CUDA 版本的 cuDNN 下載

2.解壓文件

(1) 進入下載目錄
(2) 解壓 .xz

xz -d cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz

(3) 解壓 .tar

tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar

3.替換文件

(1) 切換超級用戶

su

(2) 復制 lib 或 lib64 文件夾下的文件到 CUDA 安裝位置的 lib64 文件夾下

cp lib/* /usr/local/cuda-11.8/lib64/

(3) 復制 include 文件夾下的全部文件到 CUDA 安裝位置的 include 文件夾下

cp include/* /usr/local/cuda-11.8/include/

4.驗證安裝

cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

出現下圖提示即為安裝成功

三、切換 CUDA 版本

1.切換版本

(1) 進入 /usr/local 目錄下

cd /usr/local

(2) 刪除軟連接

sudo rm -rf cuda

(3) 新建軟連接到想切換的版本

sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda

2.檢查版本

nvcc -V

可以看到版本已經切換成功