Ubunt16.04 搭建 GPU 显卡驱动 + CUDA9.0 + cuDNN7 详细教程
紅色石頭的個人網(wǎng)站：www.redstonewill.com
如果你的電腦安裝了 Ubuntu16.04，而且電腦自帶一塊 NVIDIA GeForce 的 GPU 顯卡，那么不用來跑深度學習模型就太可惜了！關(guān)于這方面的網(wǎng)上教程很多，但大都良莠不齊。這篇文章將手把手教你如何安裝 GPU 顯卡驅(qū)動、CUDA9.0 和 cuDNN7。值得一試！
注意：由于不同系統(tǒng)，不同 GPU 對應(yīng)的 CUDA 和 cuDNN 均有差異，本文僅以 Ubuntu16.04、NVIDIA GeForce GPU、CUDA9.0 和 cuDNN7 為例，來介紹安裝過程。其它版本過程類似，但存在細微差別，請讀者以實踐為主。
必要條件
工欲善其事必先利其器，首先確定你的電腦滿足下面兩個條件。
- 安裝 Ubuntu16.04
- 電腦自帶一塊 NVIDIA GeForce GPU 顯卡
一、安裝 GPU 顯卡驅(qū)動
有一塊 NVIDIA GPU 顯卡 還不夠，還需要安裝 GPU 顯卡驅(qū)動。
1、首先登陸 GeForce 官網(wǎng)，在手動搜索驅(qū)動程序里，選擇自己對于的系統(tǒng)，注意是“Linux 64-bit”。其他選項默認即可。然后，點擊開始搜索，將會自動搜索你的電腦相匹配的 GPU 顯卡驅(qū)動，搜索結(jié)果如下圖所示：
如上圖所示，結(jié)果會顯示可用的 GPU 顯卡驅(qū)動程序，確定你需要的版本，直接點擊就可以下載了。例如我選擇的版本是418.43，下載后的文件是 NVIDIA-Linux-x86_64-418.43.run。
2、打開終端 Terminal，首先卸載系統(tǒng)之前安裝的 NVIDIA 顯卡驅(qū)動，以防萬一。（沒有安裝的可以忽略這一步）
$ sudo apt-get --purge remove nvidia-*3、打開 blacklist：
$ sudo gedit /etc/modprobe.d/blacklist.conf在最后添加幾行：
blacklist vga16fb blacklist nouveau blacklist rivafb blacklist nvidiafb blacklist rivatv這時候在終端輸入：
$ lsmod | grep nouveau如果沒有輸出，可以直接進行下一步，安裝驅(qū)動。如果仍有輸出，則在終端輸入：
$ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf直接寫入：
blacklist nouveau blacklist lbm-nouveau options nouveau modeset=0 alias nouveau off alias lbm-nouveau off保存關(guān)閉，再在終端輸入下面的命令驗證是否沒有輸出了。
$ sudo update-initramfs -u $ lsmod | grep nouveau # 如果這樣還有輸出，reboot一次到這一步，應(yīng)該沒有輸出了，可以進行安裝驅(qū)動了。
4、在正式安裝驅(qū)動之前，請記住你的驅(qū)動的 .run 文件的下載路徑和完整的文件名。按 Ctrl+Alt+F1 進入命令行界面，輸入下面的命令：
$ sudo /etc/init.d/lightdm stop $ sudo init 3進入 .run 文件的下載路徑，使用 chmod 命令增加 .run 文件的運行權(quán)限。
$ chmod a+x NVIDIA-Linux-x86_64-418.43.run $ sudo ./NVIDIA-Linux-x86_64-418.43.run -no-x-check -no-nouveau-check -no-opengl-files #這里是關(guān)閉x server，nouveau和不下載opengl文件然后就會開始安裝程序，安裝完成之后，輸入下面的命令：
$ nvidia-smi如果出現(xiàn)與下面類似的結(jié)果，證明驅(qū)動安裝成功了。
最后，回到圖像界面。
這樣，驅(qū)動就算正式安裝成功了。
二、安裝 CUDA9.0
1、首先，登錄 CUDA 的下載網(wǎng)站，由于現(xiàn)在 CUDA 的版本已經(jīng)更新到 10 了，我們需要安裝 9.0 版本。因此，點擊 Legacy Releases，進入之前的版本列表。
然后，在 CUDA 版本列表中，選擇 9.0 版本。
選擇對應(yīng)的系統(tǒng)等信息，就可以點擊 Download 進行下載了。注意選擇下載 runfile 文件。
2、打開終端，進入 CUDA9.0 的下載目錄，輸入：
然后就會進入安裝。在安裝過程中會詢問你一些配置信息，一般直接選擇默認選項就行。有一個必須注意的地方是會讓你選擇是否安裝 Graphics Driver for Linux-x86_64：XXXX 版本，這個地方必須選擇 no！否則會覆蓋之前的驅(qū)動。
3、安裝完畢之后，聲明環(huán)境變量。打開 .bashrc 文件，在終端輸入：
$ sudo gedit ~/.bashrc在文件最后加上：
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}保存退出。
$ source ~/.bashrc4、驗證 GPU 驅(qū)動版本：418.43
$ cat /proc/driver/nvidia/version驗證 CUDA 版本：9.0
$$ nvcc -V5、還有可能遇到的問題是沒有安裝 gcc 或者 gcc 版本過高需要手動進行降級，都比較好解決，這里就不再贅述了。
三、安裝 cuDNN
1、首先，進入 cuDNN 的下載網(wǎng)站。值得一提的是，下載之前是需要先注冊賬號。在下圖所示的版本列表中，選擇適配 CUDA9.0 的，對應(yīng)的是 Linux 系統(tǒng)。
2、下載完畢之后，解壓：
解壓命令輸出如下：
cuda/include/cudnn.h cuda/NVIDIA_SLA_cuDNN_Support.txt cuda/lib64/libcudnn.so cuda/lib64/libcudnn.so.7 cuda/lib64/libcudnn.so.7.6.1 cuda/lib64/libcudnn_static.a拷貝相關(guān)文件至指定目錄：
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ $ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/3、到本地 cuda 路徑，創(chuàng)建軟連接，終端輸入：
$ cd /usr/local/cuda/lib64 $ sudo chmod +r libcudnn.so.7.6.1 $ sudo ln -sf libcudnn.so.7.6.1 libcudnn.so.7 $ sudo ln -sf libcudnn.so.7 libcudnn.so $ sudo ldconfig注意，這里的文件名即版本號就參照之前解壓出來的文件名對照著寫。
4、到本地 cuda 路徑下檢測一下：
$ cd /usr/local/cuda/lib64 $ ll出現(xiàn) cuDNN 對應(yīng)的版本號就證明安裝成功！
lrwxrwxrwx 1 root root 13 3月 19 16:02 libcudnn.so -> libcudnn.so.7* lrwxrwxrwx 1 root root 17 3月 19 16:02 libcudnn.so.7 -> libcudnn.so.7.6.1* -rwxr-xr-x 1 root root 287641664 3月 19 16:00 libcudnn.so.7.6.1*大功告成！GPU 顯卡驅(qū)動 + CUDA9.0 + cuDNN7 環(huán)境配置完成。
參考文獻：
Ubuntu16.04搭建GTX1080+CUDA9.0+cuDNN7.0.5+Tensorflow1.6.0環(huán)境
總結(jié)
以上是生活随笔為你收集整理的Ubunt16.04 搭建 GPU 显卡驱动 + CUDA9.0 + cuDNN7 详细教程的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： 就在刚刚！吴恩达的这门新课程终于开放注册
- 下一篇： NET命令的用法