一、安裝 NVIDIA GPU 的驅動
二、安裝顯卡驅動
驗證 CUDA 是否安裝成功 (有root權限)
$ cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
$ sudo make
$ ./deviceQuery
如果沒有 root 權限的話,在使用者的資料夾自行下載
$ cuda-install-samples-9.0.sh ~ (安裝在自己的目錄下)
$ cd NVIDIA_CUDA-9.0_Samples/1_Utilities/deviceQuery
$ make
$ ./deviceQuery
如果 CUDA 有安裝成功的話,輸入完 ./deviceQuery 後,會出現以下的畫面
三、安裝 NVIDIA cuDNN
驗證是否有安裝成功
$ nvcc -V
如果有安裝成功的話,會出現以下畫面