Ubuntu 20.04 CUDA 11.1 cuDNN 8.0.4 install

JianJie

Nov 2, 2020

目前環境：

Ubuntu version : 20.04.1

Kernel version : 5.4.0-52

GPU: NVIDIA RTX 3080

1.顯示卡驅動安裝

在NVIDIA官網下載相對應顯卡的驅動，這裡安裝驅動 455.23版

驅動程式下載

Download drivers for NVIDIA products including GeForce graphics cards, nForce motherboards, Quadro workstations, and…

www.nvidia.com.tw

下載完成後，執行

sudo sh ./NVIDIA-Linux-x86_64–455.23.04.run

重新開機後，確認顯卡驅動，執行

nvidia-smi

2.CUDA 11.1安裝

CUDA 是 NVIDIA 研發的平行運算平台及編程模型，可利用繪圖處理單元 (GPU) 的能力大幅提升運算效能。在矩陣運算方面比起CPU，擁有更好的運算能力

按照NVIDIA官網方式安裝

CUDA Toolkit 11.1 Update 1 Downloads

Please Note: Due to an incompatibility issue, we advise users to defer updating to Linux Kernel 5.9+ until mid-November…

developer.nvidia.com

開啟Terminal並執行:

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pinsudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600wget https://developer.download.nvidia.com/compute/cuda/11.1.1/local_installers/cuda-repo-ubuntu2004-11-1-local_11.1.1-455.32.00-1_amd64.debsudo dpkg -i cuda-repo-ubuntu2004–11–1-local_11.1.1–455.32.00–1_amd64.debsudo apt-key add /var/cuda-repo-ubuntu2004–11–1-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

3.cuDNN 8.0.4安裝

cuDNN是NVIDIA打造的針對深度神經網絡的加速library，是一個用於深層神經網絡的GPU library。如果你要用GPU訓練模型，cuDNN不是必須的，但是一般會採用這個library。

到NVIDIA官網下載cuDNN (需要下載相對應的cuDNN版本）

NVIDIA cuDNN

NVIDIA cuDNN The NVIDIA CUDA® Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep…

developer.nvidia.com

這裡下載cuDNN v8.0.4，對應CUDA11.1

解壓縮檔案後，把裡面的資料複製到指定的資料夾

sudo cp cuda/include/cudnn.h /usr/lib/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/lib/cuda/lib64/
sudo chmod a+r /usr/lib/cuda/include/cudnn.h /usr/lib/cuda/lib64/libcudnn*

4.加入環境變數

sudo nano ~/.bashrc

在檔案最下方加入

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

執行

source ~/.bashrc