應用場景:Ubuntu多用戶原來共享相同的CUDA和cuDNN,現在當前用戶下有其它版本的CUDA和cuDNN使用需求。假設原來裝的是cuda-10.2并共享所有用戶,現在某一個用戶要用到cuda-11.3。
1.安裝cuda
首先下載cuda-11.3的對應版本:
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
待安裝界面出現后:
安裝界面
因為已經有顯卡驅動了,所以這里一定要把安裝驅動這一項取消!
選中Install選項后會跳轉到這一界面:
image.png
提示系統中已經存在一個cuda版本在 /usr/local/cuda 目錄下,是否更新。這里選擇 【No】,然后等幾分鐘就安裝好了。
待安裝cuda-11.3成功后,執行命令:
sudo vim ~/.bashrc
打開當前用戶環境變量界面,并在末尾換行,添加以下語句:
export PATH="$PATH:/usr/local/cuda-11.3/bin"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-11.3/lib64/"
export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda-11.3/lib64"
保存并退出。【注意】將其中的11.3改為自己對應的版本!
執行命令:
source ~/.bashrc
激活環境變量。至此,cuda就安裝完成了。接下來繼續安裝cudnn。
2.安裝cuDNN
下載匹配cuda-11.3的cuDNN版本,執行解壓命令:
tar -zvxf cudnn-11.3-linux-x64-v8.2.1.32.tgz
解壓cuDNN。【注意】:對應自己的cudnn版本!
解壓后會在當前目錄下出現一個名為cuda的文件夾,即為解壓后的cuDNN.
執行命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
將cudnn的頭文件等復制到對應目錄下,安裝完畢。
這里的不同之處在于,安裝第一個cuda版本時,通常會在 /usr/local 目錄下創建一個名為 cuda 的軟鏈接,并將 cudnn 的相應文件復制到該軟鏈接 cuda 目錄下。
而第二次安裝 cuda 時并沒有創建軟鏈接,而是安裝在 /usr/local/cuda-11.3 路徑下,并將 cudnn 文件直接復制到相應的安裝目錄 /usr/local/cuda-11.3 下即可。