前言(碎碎念)
想當(dāng)年在實(shí)驗(yàn)室就在自己電腦(雙系統(tǒng))和服務(wù)器上都搭建這個(gè)環(huán)境(Ubuntu18.04+conda+pycharm+cuda+cudnn完整流程)
還寫(xiě)了操作文檔,主要遇到了太多問(wèn)題(比如NVIDIA驅(qū)動(dòng)安裝后黑屏問(wèn)題,真是想用linux之父的話來(lái)說(shuō)一句)
現(xiàn)在再次遇到重裝服務(wù)器這個(gè)事情,又遇到了一些新坑(好多東西變了)。這次就寫(xiě)在這里吧(碎碎念完結(jié))
一、CUDA版本選擇(授予以魚(yú)不如授人以漁)
- 打開(kāi)英偉達(dá)的cuda文檔頁(yè)面 cuda文檔頁(yè)面
- 點(diǎn)這里的Release Notes(cuda toolkit頁(yè)面)
-
這里查看信息
-
兼容性
CUDA有最小支持的驅(qū)動(dòng)版本的要求,高版本的CUDA不支持低版本的驅(qū)動(dòng);驅(qū)動(dòng)版本向后兼容,即驅(qū)動(dòng)可以升級(jí)而安裝的CUDA不用升級(jí)。
-
當(dāng)前支持的cuda版本查詢(需要安裝對(duì)應(yīng)的顯卡驅(qū)動(dòng))
nvidia-smi
這里可以看到當(dāng)前的驅(qū)動(dòng)版本和cuda支持的版本(上圖的上部)
PS:這里指當(dāng)前驅(qū)動(dòng)最大支持的cuda版本,可以選擇升級(jí)驅(qū)動(dòng)或者安裝的cuda版本低于這個(gè)
-
驅(qū)動(dòng)安裝(小白提示)
另外,使用這個(gè)命令的前提條件是已經(jīng)安裝了NVIDIA的顯卡驅(qū)動(dòng)了
PS:Ubuntu可通過(guò)軟件和更新附加驅(qū)動(dòng)選項(xiàng),一鍵式安裝驅(qū)動(dòng)(如下圖)
二、CUDA安裝(本文是deb安裝)
-
兩種安裝方式
- 用deb方式安裝CUDA,會(huì)附帶安裝顯卡驅(qū)動(dòng),默認(rèn)會(huì)安裝(本文)
- 用run方式安裝CUDA,會(huì)附帶安裝顯卡驅(qū)動(dòng),可選不安裝(出門搜其他人的文章喲)
-
進(jìn)入頁(yè)面
Release Notes(文檔頁(yè)面)
這次點(diǎn)linux安裝指南
頁(yè)面有很細(xì)節(jié)說(shuō)明,這里就不贅述了,乖寶寶們自己去看喲(跳轉(zhuǎn)鏈接如下)
下載頁(yè)面 cuda下載頁(yè)面
-
具體選項(xiàng)
然后頁(yè)面下方會(huì)刷出具體的命令,一行一行按照它的命令執(zhí)行操作就好
-
驗(yàn)證安裝結(jié)果
nvcc -V
如果沒(méi)有出現(xiàn)就是沒(méi)有配置cuda的環(huán)境變量,輸入下面的命令
sudo nano ~/.bashrc
在末尾加上以下兩行路徑信息
export PATH="/usr/local/{cuda-12.0}/bin:$PATH" export LD_LIBRARY_PATH="/usr/local/{cuda-12.0}/lib64:$LD_LIBRARY_PATH"
PS:{}是你自己的路徑,不一定是cuda-12.0
查看具體的版本的命令如下cd /usr/local/ ls
路徑信息無(wú)誤后,ctrl+x保存并退出,此時(shí)需要刷新
source ~/.bashrc nvcc -V
三、CUDNN安裝
-
頁(yè)面搜索
搜索引擎搜cudnn,官網(wǎng)的首頁(yè)如下圖,鏈接cudnn頁(yè)面
接下來(lái)需要你登錄(沒(méi)有賬號(hào)的話需要你注冊(cè)一個(gè)NVIDIA的賬號(hào),這個(gè)我就愛(ài)莫能助了)之后,點(diǎn)同意,并下載
文檔頁(yè)面點(diǎn)開(kāi)后,繼續(xù)點(diǎn)安裝向?qū)А_@里給出安裝向?qū)ы?yè)面的鏈接cudnn說(shuō)明頁(yè)面
-
安裝命令
因?yàn)閏uda等已經(jīng)搞定了,這里直接跳轉(zhuǎn)到對(duì)應(yīng)的安裝方式這里。
總共是7步,這里就沒(méi)有截全了,對(duì)應(yīng)的命令放下面了
A.先進(jìn)入cudnn的deb下載路徑($distro 為ubuntu1804、ubuntu2004, ubuntu2204等),tab直接自動(dòng)補(bǔ)全,輸入下面命令
sudo dpkg -i cudnn-local-repo-$distro-8.x.x.x_1.0-1_amd64.deb
B.導(dǎo)入CUDA GPG key
sudo cp /var/cudnn-local-repo-*/cudnn-local-*-keyring.gpg /usr/share/keyrings/
C.刷新元數(shù)據(jù)
sudo apt-get update
D.分別安裝三個(gè)庫(kù)
sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y
再次說(shuō)明x.x.x是對(duì)應(yīng)下載的cudnn版本信息,X.Y是 cuda的大版本和小版本信息
上面步驟D,可能會(huì)發(fā)生錯(cuò)誤,說(shuō)找不到,解決方法如下
進(jìn)入/var對(duì)應(yīng)的cudnn-local-repo-ubuntu2004-8.x.x.x文件中
使用dpkg命名安裝上圖路徑的三個(gè)包或者
修改4中的cuda版本,我這里是12.0但包實(shí)際的版本是12.2.修改命令后就可以了
sudo apt-get install libcudnn8=8.9.7.29-1+cuda12.2 sudo apt-get install libcudnn8-dev=8.9.7.29-1+cuda12.2 sudo apt-get install libcudnn8-samples=8.9.7.29-1+cuda12.2
-
驗(yàn)證
借用網(wǎng)上的信息,如何檢測(cè)版本信息,舊cudnn版本用下面第1個(gè),新的用第2個(gè)
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
不過(guò)新的已經(jīng)可能上面的路徑中了,解決方法如下,輸入下面的命令,找到真實(shí)的位置
where is cudnn_version.h
然后再根據(jù)這個(gè)路徑信息輸入下面任意兩個(gè)命令文章來(lái)源:http://www.zghlxwxcb.cn/news/detail-769277.html
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 5 MAJOR
文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-769277.html
-
到了這里,關(guān)于Ubuntu 20.04(linux) cuda(12)+cudnn的deb方式安裝以及驗(yàn)證(寶寶也適用喲)的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!