準備工作

1.檢視GPU是否支援CUDA

lspci | grep -i nvidia

2.檢視Linux版本

uname -m && cat /etc/*release

nvidia驅動

1. 先解除安裝原有N卡驅動

#for case1: original driver installed by apt-get:
sudo apt-get remove --purge nvidia*
#for case2: original driver installed by runfile:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall

2. 禁用nouveau驅動

sudo gedit /etc/modprobe.d/blacklist.conf

在文字最後新增：

blacklist nouveau
options nouveau modeset=0

然後執行：

sudo update-initramfs -u

重啟之後，可以檢視nouveau有沒有執行:

lsmod | grep nouveau # 沒輸出代表禁用生效

3.禁用X-Window服務

sudo service lightdm stop #這會關閉圖形介面

按Ctrl-Alt+F1進入命令列介面，輸入使用者名稱和密碼登入。

4.命令列安裝驅動

驅動網址https://www.nvidia.cn/Download/index.aspx?lang=cn

#給驅動run檔案賦予執行許可權：
sudo chmod +x NVIDIA-Linux-x86_64-384.59.run
#後面的引數非常重要，不可省略：
sudo ./NVIDIA-Linux-x86_64-384.59.run ?no-opengl-files

5.Driver測試：

nvidia-smi #若列出GPU的資訊列表，表示驅動安裝成功
nvidia-settings #若彈出設定對話方塊，亦表示驅動安裝成功

CUDA安裝

1.下載對應系統的CUDA

網址http://developer.nvidia.com/cuda-downloads 選擇runfile安裝

2.安裝

sudo sh cuda_<version>_linux.run

開始安裝之後，需要閱讀說明，可以使用Ctrl + C直接閱讀完成，或者使用空格鍵慢慢閱讀。下面為安裝選項：

（是否同意條款，必須同意才能繼續安裝）
accept/decline/quit: accept
（這裡不要安裝驅動，因為已經安裝最新的驅動了，否則可能會安裝舊版本的顯示卡驅動，導致重複登入的情況）
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: n
Install the CUDA 10.0 Toolkit?（是否安裝CUDA 10 ，這裡必須要安裝）
(y)es/(n)o/(q)uit: y
Enter Toolkit Location（安裝路徑，使用預設，直接回車就行）
[ default is /usr/local/cuda-10.0 ]: 
Do you want to install a symbolic link at /usr/local/cuda?（同意建立軟連結）
(y)es/(n)o/(q)uit: y
Install the CUDA 10.0 Samples?（不用安裝測試，本身就有了）
(y)es/(n)o/(q)uit: n
Installing the CUDA Toolkit in /usr/local/cuda-10.0 ...（開始安裝）

3.配置環境變數

sudo gedit ~/.bashrc

末尾加入

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

更新

source ~/.bashrc

4.驗證安裝

檢視cuda版本

nvcc -V
CUDA Sample測試：
#編譯並測試裝置 deviceQuery：
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
make
./deviceQuery
#編譯並測試頻寬 bandwidthTest：
cd ../bandwidthTest
make
./bandwidthTest

如果這兩個測試的最後結果都是Result = PASS，說明CUDA安裝成功。

*5.CUDA解除安裝

在命令列中輸入

sudo apt-get remove cuda 
sudo apt-get autoclean
sudo apt-get remove cuda*

然後在目錄切換到usr/local/下

cd /usr/local/
sudo rm -r cuda-9.1

cuDNN安裝

下載對應版本cuDNN https://developer.nvidia.com/cudnn

tar xvzf cudnn-9.2-linux-x64-v7.1
sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo ldconfig

Reference

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions

總結

以上所述是小編給大家介紹的Ubuntu16.04安裝nvidia驅動+CUDA+cuDNN的教程,希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回覆大家的。在此也非常感謝大家對碼農教程網站的支援！
如果你覺得本文對你有幫助，歡迎轉載，煩請註明出處，謝謝！