深度學習圖形工作站環境安裝與配置(1)CUDA和cuDNN的安裝

介紹

筆者實驗室的圖形工作站基於Ubuntu 18.04,請先安裝Ubuntu操作系統。推薦安裝Ubuntu 18.04 LTS,LTS意味着Long Term Support,你可以在官網下載並刻錄U盤安裝。Ubuntu操作系統官網下載,可以看到該系統支持到2023年4月,是較爲穩定的版本。

顯卡驅動的安裝

如果你確定你的顯卡驅動被正確的安裝了,可以跳過這一步
在剛裝好Ubuntu操作系統時,默認的顯卡驅動是nouveau,這是一種開源但低效的驅動,你需要更換成對應版本的驅動。

第一步:查看顯卡版本,獲取安裝包

lshw -numeric -C display

顯卡信息
在這裏可以看到一串十六進制數字,這好像不是我們熟悉的顯卡型號,實際上,這是這是顯卡型號的16進製表達,你需要解碼其中的含義,可以去這個網站,在那個下面的空白框中輸入這串數字(如我的就是1e81),你就可以得到顯卡型號。
顯卡
接下來就是在nvdia官網下載對應的驅動,在這裏選擇你的顯卡型號,和對應的linux及語言等。
顯卡
搜索之後,你可以進一步滑下來查看是否支持你的顯卡
顯卡
此時一般會彈出下載鏈接,記住他,同時記住安裝包的名稱,然後執行,

sudo  apt-get install make
#下面的鏈接應該是你獲取到的
sudo wget https://cn.download.nvidia.cn/XFree86/Linux-x86_64/440.82/NVIDIA-Linux-x86_64-440.82.run

第二步 安裝

首先需要檢查nouveau是否被正確卸載,執行

sudo apt-get install vim
vim /etc/modprobe.d/blacklist.conf

在其中加入一行命令來禁用nouveau

blacklist nouveau

重啓系統

reboot

執行下列命令,如果沒有任何輸出,則禁用成功

lsmod | grep nouveau

轉爲文本命令窗口,準備安裝

sudo telinit 3

進入安裝包所在目錄,執行

bash NVIDIA-Linux-x86_64-440.82.run

按着步驟安裝就行了

CUDA的安裝

第一步:檢查安裝環境

進入cuda官網,選擇對應版本,他會給出安裝的指令,比如我的就是

wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run

值得注意的是,你的系統環境需要滿足一定的條件,詳見官方文檔
在這裏插入圖片描述
最常見的一些錯誤就是,顯卡驅動仍然是默認的nouveau,詳見第一步
二是gcc版本太高,比如圖中可以看出Ubuntu18.04下gcc不能高於7.3.0,此時需要降級
你可以選擇很多種方式,如果想要剛好安裝gcc7.3比較麻煩,但你可以用apt來安裝

sudo apt-get install gcc-6
cd /usr/bin
ls

此時你會看見bin目錄下的已經安裝的gcc
在這裏插入圖片描述
你需要將gcc-6鏈接成gcc,首先刪除原來的軟連接,將gcc-6鏈接上去,執行

sudo rm -rf gcc
sudo ln -s gcc-6 gcc

第二步:安裝

執行你上次得到的命令執行即可,圖爲安裝完成,這還沒有安裝成功的,需要進一步操作
在這裏插入圖片描述
重新啓動系統,進入終端

reboot
#重新連接

執行以下幾行,這裏的路徑應該是你的安裝路徑

export PATH="/usr/local/cuda-10.2/bin:$PATH" 
export LD_LIBRARY_PATH="/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH" 

再執行

nvcc -V

可以看到這樣的信息就說明成功了!
在這裏插入圖片描述
你還可以編譯samples來檢查是否正常安裝,這裏不多敘述了

cuDNN的安裝

根據你的cuda版本選擇對應的cuDNN,在官網選擇你需要的版本,記下下載鏈接,用wget命令下載並解壓
在這裏插入圖片描述

wget https://developer.download.nvidia.cn/compute/machine-learning/cudnn/secure/7.6.4.38/Production/10.1_20190923/cudnn-10.1-linux-x64-v7.6.4.38.tgz?If1F9NslzoB6a4LmyJhaQpBONTd2SZjQod-58RnXpsEHYrwZU0dMUgR9KDav1Ie07nzTQgX5s8eL3EnuBGACBEk-K2o-Kg3DUXjeWllNjIi6U9miGGsxm4a48bgHb2c5otF5uhRtR6jPKcQtnLksbXDvleDWrFDzhX-yu6s7ReA5DPnOr2UhDx3MOKj0nxjkKISQ3GLJrRQadOIXmsvRsnWoeMaR7bC64A --no-check-certificate
tar -zxvf cudnn-10.1-linux-x64-v7.6.4.38.tgz

在這裏插入圖片描述
將cuda文件夾(剛解壓出來的那個)中相應文件拷貝到相應的cuda安裝文件夾下
並更改其權限,安裝工具庫

sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo cp  cuda/include/cudnn.h /usr/local/cuda-10.2/include/
sudo chmod 777 /usr/local/cuda-10.2/lib64/libcudnn*
sudo chmod 777 /usr/local/cuda-10.2/include/cudnn.h
sudo apt-get install libcupti-dev
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章