ubuntu22.0.4下CUDA12.1和cuDNN8.9.1的安裝

z_s_s發表於2024-12-03
CUDA
命令查詢地址:https://developer.nvidia.com/cuda-downloads
Terminal(快捷鍵Ctrl+Alt+t)裡輸入一下命令:

wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run

sudo sh cuda_12.1.0_530.30.02_linux.run

馬上顯示一個問題,Do you accept the above EULA?(accept/decline/quit):
輸入:accept

然後會有一個CUDA Installer 的多項選擇,如果裝過了Driver則回車改選項去掉,預設選擇CUDA Toolkit、CUDA Demo、CUDA Documentation。
接著,移動到Install安裝。

接下來配置環境變數:
vim ~/.bashrc

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

Vim 中按Esc 再按:,輸入wq (write and quit)

重新整理檔案

source ~/.bashrc

輸入nvcc-V

驗證安裝

cuDNN
官網(https://developer.nvidia.com/cudnn-downloads)登陸註冊,下載對應版本的deb檔案。

sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.1.23_1.0-1_amd64.deb

報錯:

The public cudnn-local-repo-ubuntu2204-8.9.1.23 GPG key does not appear to be installed. To install the key, run this command: sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.1.23/cudnn-local-E7A7D88D-keyring.gpg /usr/share/keyrings/

按照提示,你需要安裝 cuDNN 倉庫的 GPG 金鑰。執行以下命令來安裝金鑰:

sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.1.23/cudnn-local-E7A7D88D-keyring.gpg /usr/share/keyrings/

然後,你可以更新包列表:

sudo apt update

安裝完 GPG 金鑰後,接下來需要安裝 cuDNN 包。執行以下命令來安裝 cuDNN:

sudo apt install libcudnn8 libcudnn8-dev

只安裝 libcudnn8libcudnn8-dev: 通常,libcudnn8libcudnn8-dev 已經包含了大部分需要的功能,而 libcudnn8-doc 只是附帶的文件包。你可以跳過 libcudnn8-doc 包,直接安裝 libcudnn8libcudnn8-dev

如果一切順利,安裝完成後,cudnn_version.h 檔案應該會出現在 /usr/include/ 目錄下。你可以透過以下命令檢查安裝:

cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
 

相關文章