Ubuntu多用戶實現(xiàn)不同版本的CUDA和cuDNN安裝

應用場景:Ubuntu多用戶原來共享相同的CUDA和cuDNN,現(xiàn)在當前用戶下有其它版本的CUDA和cuDNN使用需求。假設原來裝的是cuda-10.2并共享所有用戶,現(xiàn)在某一個用戶要用到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

待安裝界面出現(xiàn)后:


安裝界面

因為已經(jīng)有顯卡驅(qū)動了,所以這里一定要把安裝驅(qū)動這一項取消!

選中Install選項后會跳轉(zhuǎn)到這一界面:


image.png

提示系統(tǒng)中已經(jīng)存在一個cuda版本在 /usr/local/cuda 目錄下,是否更新。這里選擇 【No】,然后等幾分鐘就安裝好了。

待安裝cuda-11.3成功后,執(zhí)行命令:

sudo vim ~/.bashrc

打開當前用戶環(huán)境變量界面,并在末尾換行,添加以下語句:

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改為自己對應的版本!

執(zhí)行命令:

source ~/.bashrc

激活環(huán)境變量。至此,cuda就安裝完成了。接下來繼續(xù)安裝cudnn。

2.安裝cuDNN

下載匹配cuda-11.3的cuDNN版本,執(zhí)行解壓命令:

tar -zvxf cudnn-11.3-linux-x64-v8.2.1.32.tgz

解壓cuDNN。【注意】:對應自己的cudnn版本!

解壓后會在當前目錄下出現(xiàn)一個名為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的頭文件等復制到對應目錄下,安裝完畢。

這里的不同之處在于,安裝第一個cuda版本時,通常會在 /usr/local 目錄下創(chuàng)建一個名為 cuda 的軟鏈接,并將 cudnn 的相應文件復制到該軟鏈接 cuda 目錄下。

而第二次安裝 cuda 時并沒有創(chuàng)建軟鏈接,而是安裝在 /usr/local/cuda-11.3 路徑下,并將 cudnn 文件直接復制到相應的安裝目錄 /usr/local/cuda-11.3 下即可。

?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容