1.安裝顯示卡驅動

簡單一步在ubuntu18.04下nvidia driver的安裝

直接通過Ubuntu自帶UI介面安裝，應用更改自動安裝，重啟生效

解除安裝CUDA和cuDNN看這

2.安裝CUDA10.1

官網下載地址

執行.run檔案

sudo sh cuda_10.1.105_418.39_linux.run

• 1

因為已經安裝了NVIDIA 顯示卡驅動，所以這裡不勾選安裝驅動

等待片刻，出現安裝成功

新增環境變數

sudo vim ~/.bashrc

• 1

export CUDA_HOME=/usr/local/cuda-10.1
export PATH=/usr/local/cuda-10.1/bin${PATH:+:$PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
 

• 1
• 2
• 3

驗證

nvcc --version

• 1

3.安裝cudnn7.6.2

官網下載地址

下載cuDNN Library for Linux，解壓，在解壓的資料夾中開啟終端
複製檔案到目錄

sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/

• 1
• 2

建立軟連結
在/usr/local/cuda/lib64目錄下開啟終端，執行如下指令：

sudo chmod +r libcudnn.so.7.6.2
sudo ln -sf libcudnn.so.7.6.2 libcudnn.so.7  
sudo ln -sf libcudnn.so.7 libcudnn.so     
sudo ldconfig
 

• 1
• 2
• 3
• 4

檢視當前cudnn的版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

• 1