應(yīng)用場(chǎng)景:Ubuntu多用戶(hù)原來(lái)共享相同的CUDA和cuDNN痰催,現(xiàn)在當(dāng)前用戶(hù)下有其它版本的CUDA和cuDNN使用需求兜辞。假設(shè)原來(lái)裝的是cuda-10.2并共享所有用戶(hù),現(xiàn)在某一個(gè)用戶(hù)要用到cuda-11.3夸溶。
1.安裝cuda
首先下載cuda-11.3的對(duì)應(yīng)版本:
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run
然后進(jìn)行安裝:
sudo sh cuda_11.3.1_465.19.01_linux.run
待安裝界面出現(xiàn)后:
因?yàn)橐呀?jīng)有顯卡驅(qū)動(dòng)了逸吵,所以這里一定要把安裝驅(qū)動(dòng)這一項(xiàng)取消!
選中Install選項(xiàng)后會(huì)跳轉(zhuǎn)到這一界面:
提示系統(tǒng)中已經(jīng)存在一個(gè)cuda版本在 /usr/local/cuda 目錄下缝裁,是否更新扫皱。這里選擇 【No】,然后等幾分鐘就安裝好了捷绑。
待安裝cuda-11.3成功后韩脑,執(zhí)行命令:
sudo vim ~/.bashrc
打開(kāi)當(dāng)前用戶(hù)環(huán)境變量界面,并在末尾換行粹污,添加以下語(yǔ)句:
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改為自己對(duì)應(yīng)的版本!
執(zhí)行命令:
source ~/.bashrc
激活環(huán)境變量壮吩。至此进苍,cuda就安裝完成了。接下來(lái)繼續(xù)安裝cudnn鸭叙。
2.安裝cuDNN
下載匹配cuda-11.3的cuDNN版本觉啊,執(zhí)行解壓命令:
tar -zvxf cudnn-11.3-linux-x64-v8.2.1.32.tgz
解壓cuDNN。【注意】:對(duì)應(yīng)自己的cudnn版本递雀!
解壓后會(huì)在當(dāng)前目錄下出現(xiàn)一個(gè)名為cuda的文件夾柄延,即為解壓后的cuDNN.
執(zhí)行命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.3/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.3/lib64
將cudnn的頭文件等復(fù)制到對(duì)應(yīng)目錄下蚀浆,安裝完畢缀程。
這里的不同之處在于搜吧,安裝第一個(gè)cuda版本時(shí),通常會(huì)在 /usr/local 目錄下創(chuàng)建一個(gè)名為 cuda 的軟鏈接杨凑,并將 cudnn 的相應(yīng)文件復(fù)制到該軟鏈接 cuda 目錄下滤奈。
而第二次安裝 cuda 時(shí)并沒(méi)有創(chuàng)建軟鏈接,而是安裝在 /usr/local/cuda-11.3 路徑下撩满,并將 cudnn 文件直接復(fù)制到相應(yīng)的安裝目錄 /usr/local/cuda-11.3 下即可蜒程。