上篇記錄了Ubuntu下安裝INVIDIA顯示卡驅動的方法，儘管可以選擇CUDA自帶的驅動，但為了避免不必要的問題，儘量單獨安裝。

如果沒有單獨安裝驅動，建議多找幾篇部落格，對比來看，瞭解有哪些坑，才能有效避免入坑。

下文為安裝步驟

STEP1：CUDA官網選擇適合自己系統的版本下載。

我的系統是Ubuntu18.04、64位，選擇CUDA10版本如下：

 STEP2：安裝CUDA

在命令列中鍵入：

sudo sh cuda_10.0.130_410.48_linux.run

一直按Enter直至把宣告讀完（太多了吧！）

如果驅動是獨立安裝了，一定要選擇不安裝驅動！選擇如下：

會提示：

***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 384.00 is required for CUDA 10.0 functionality to work.

To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:

sudo <CudaInstaller>.run -silent -driver

這是因為我們在安裝CUDA的時候沒有選擇安裝驅動，提示需要安裝驅動，忽略就行。

STEP3：新增環境變數：

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

儲存並退出

source ~/.bashrc

STEP4：測試是否安裝成功

上面的選項選擇安裝了CUDA例子，執行其中一個來測試是否安裝成功：

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

結果如下就表明成功啦

STEP5：安裝CUDNN

在官網下載安裝包，需要註冊登入才能下載。選擇適合自己的版本

 下載完成後解壓並進入資料夾：

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

在終端檢視CUDNN版本：

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

完成！