更新
自己在配置的時候發(fā)現(xiàn)勿璃,cuda與cudnn版本相差不大也可以混著用,不一定嚴(yán)絲合縫的匹配。
安裝教程針對cuda11.5以上的版本随珠,以下的版本需自己翻看之前人寫的教程。
安裝方法直接參考用戶手冊安裝
Installation Guide - NVIDIA Docs
1.3.2. Debian Local Installation
Download the Debian local repository installation package. Before issuing the following commands, you must replace X.Y and 8.x.x.x with your specific CUDA and cuDNN versions.
- Navigate to your <cudnnpath> directory containing the cuDNN Debian local installer file.
- Enable the local repository.
sudo dpkg -i cudnn-local-repo-\${OS}-8.x.x.x_1.0-1_amd64.deb
or
sudo dpkg -i cudnn-local-repo-\${OS}-8.x.x.x_1.0-1_arm64.deb
執(zhí)行完這個命令之后會在/var/目錄下出現(xiàn)一個同樣名字的文件夾猬错。5窗看,6,7步的操作是基于這個文件夾中的.deb文件倦炒。
- Import the CUDA GPG key.
sudo cp/var/cudnn-local-repo-\*/cudnn-local-\*-keyring.gpg /usr/share/keyrings/
這個命令在執(zhí)行第二步時會自動提供显沈。復(fù)制粘貼即可
- Refresh the repository metadata.
sudo apt-get update
如果第二步安裝錯誤,這里會有報錯逢唤。卸載掉上次安裝錯誤的包后錯誤消失拉讯。卸載方法參考文章最后。
- Install the runtime library.
sudo apt-get install libcudnn8=8.x.x.x-1+cudaX.Y
- Install the developer library.
sudo apt-get install libcudnn8-dev=8.x.x.x-1+cudaX.Y
- Install the code samples.
sudo apt-get install libcudnn8-samples=8.x.x.x-1+cudaX.Y
但是有個坑要注意:
在安裝cudnn時鳖藕,必須根注意cudnn的版本要跟cuda的版本匹配遂唧。(cuda的版本是敲nvcc -V時出現(xiàn)的版本號,而不是nvidia-smi出現(xiàn)的版本號)
如果用比較舊的cuda版本吊奢,不能直接用NVIDIA的最新cudnn包盖彭,因為此時的包中是對應(yīng)最新的cuda版本纹烹。需要翻看之前的cudnn包。
但是你不知道cudnn包中對應(yīng)哪個cuda版本召边,一般需要自己dpkg 安裝一下铺呵,然后在/var目錄下找到對應(yīng)的包,從而找到適合自己的cudnn版本隧熙。
如:
點進(jìn)去就能看到此時cudnn中對應(yīng)的哪種cuda版本片挂。
如果此時目錄中沒有對應(yīng)自己cuda版本的deb,那就重新下載一個新的cudnn包贞盯。
我總結(jié)一下, 可能不準(zhǔn)音念,以實際操作為準(zhǔn):
cudnn-local-repo-ubuntu2004-8.8.1.3_1.0-1_amd64.deb --> cuda11.8
cudnn-local-repo-ubuntu2004-8.8.0.121_1.0-1_amd64.deb --> cuda11.8
cudnn-local-repo-ubuntu2004-8.7.0.84_1.0-1_amd64.deb --> cuda11.8
cudnn-local-repo-ubuntu2004-8.6.0.163_1.0-1_amd64.deb --> cuda11.8
cudnn-local-repo-ubuntu2004-8.5.0.96_1.0-1_amd64.deb --> cuda11.7
cudnn-local-repo-ubuntu2004-8.4.1.50_1.0-1_amd64.deb --> cuda11.6
那之前安裝錯的包記得刪除。用下面的命令刪除即可:
apt-get --purge remove <package> # 刪除軟件及其配置文件
<package> 就是上面第二步安裝的包的名稱躏敢,即cudnn-local-repo-${OS}-8.x.x.x_1.0-1_amd64(記得替換${OS}-8.x.x.x)
或者先用dpkg -l | grep cudnn
找到cudnn軟件名稱闷愤,然后執(zhí)行上述命令。