點擊上方“AI有道”，選擇“星標”公眾號

重磅干貨，第一時間送達

如果你的電腦安裝了 Ubuntu16.04，而且電腦自帶一塊 NVIDIA GeForce 的 GPU 顯卡，那么不用來跑深度學習模型就太可惜了！關于這方面的網上教程很多，但大都良莠不齊。這篇文章將手把手教你如何安裝 GPU 顯卡驅動、CUDA9.0 和 cuDNN7。值得一試！

注意：由于不同系統，不同 GPU 對應的 CUDA 和 cuDNN 均有差異，本文僅以 Ubuntu16.04、NVIDIA GeForce GPU、CUDA9.0 和 cuDNN7 為例，來介紹安裝過程。其它版本過程類似，但存在細微差別，請讀者以實踐為主。

必要條件

工欲善其事必先利其器，首先確定你的電腦滿足下面兩個條件。

• 安裝 Ubuntu16.04

• 電腦自帶一塊 NVIDIA GeForce GPU 顯卡

一、安裝 GPU 顯卡驅動

有一塊 NVIDIA GPU 顯卡 還不夠，還需要安裝 GPU 顯卡驅動。

1、首先登陸 GeForce 官網：

https://www.geforce.cn/drivers

在手動搜索驅動程序里，選擇自己對于的系統，注意是“Linux 64-bit”。其他選項默認即可。然后，點擊開始搜索，將會自動搜索你的電腦相匹配的 GPU 顯卡驅動，搜索結果如下圖所示：

如上圖所示，結果會顯示可用的 GPU 顯卡驅動程序，確定你需要的版本，直接點擊就可以下載了。例如我選擇的版本是418.43，下載后的文件是 NVIDIA-Linux-x86_64-418.43.run。

2、打開終端 Terminal，首先卸載系統之前安裝的 NVIDIA 顯卡驅動，以防萬一。（沒有安裝的可以忽略這一步）

$ 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

如果沒有輸出，可以直接進行下一步，安裝驅動。如果仍有輸出，則在終端輸入：

$ sudo gedit /etc/modprobe.d/blacklist-nouveau.conf

直接寫入：

blacklist nouveau blacklist lbm-nouveau options nouveau modeset=0 alias nouveau off alias lbm-nouveau off

保存關閉，再在終端輸入下面的命令驗證是否沒有輸出了。

$ sudo update-initramfs -u $ lsmod | grep nouveau # 如果這樣還有輸出，reboot一次

到這一步，應該沒有輸出了，可以進行安裝驅動了。

4、在正式安裝驅動之前，請記住你的驅動的 .run 文件的下載路徑和完整的文件名。按 Ctrl+Alt+F1 進入命令行界面，輸入下面的命令：

$ sudo /etc/init.d/lightdm stop $ sudo init 3

進入 .run 文件的下載路徑，使用 chmod 命令增加 .run 文件的運行權限。

$ 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 #這里是關閉x server，nouveau和不下載opengl文件

然后就會開始安裝程序，安裝完成之后，輸入下面的命令：

$ nvidia-smi

如果出現與下面類似的結果，證明驅動安裝成功了。

最后，回到圖像界面。

$ sudo /etc/init.d/lightdm start

這樣，驅動就算正式安裝成功了。

二、安裝 CUDA9.0

1、首先，登錄 CUDA 的下載網站：

https://developer.nvidia.com/cuda-downloads

由于現在 CUDA 的版本已經更新到 10 了，我們需要安裝 9.0 版本。因此，點擊 Legacy Releases，進入之前的版本列表。

然后，在 CUDA 版本列表中，選擇 9.0 版本。

選擇對應的系統等信息，就可以點擊 Download 進行下載了。注意選擇下載 runfile 文件。

2、打開終端，進入 CUDA9.0 的下載目錄，輸入：

$ chmod a+x cuda_9.0.176_384.81_linux.run $ sudo ./cuda_9.0.176_384.81_linux.run

然后就會進入安裝。在安裝過程中會詢問你一些配置信息，一般直接選擇默認選項就行。有一個必須注意的地方是會讓你選擇是否安裝 Graphics Driver for Linux-x86_64：XXXX 版本，這個地方必須選擇 no！否則會覆蓋之前的驅動。

3、安裝完畢之后，聲明環境變量。打開 .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 ~/.bashrc

4、驗證 GPU 驅動版本：418.43

$ cat /proc/driver/nvidia/version

驗證 CUDA 版本：9.0

$$ nvcc -V

5、還有可能遇到的問題是沒有安裝 gcc 或者 gcc 版本過高需要手動進行降級，都比較好解決，這里就不再贅述了。

三、安裝 cuDNN

1、首先，進入 cuDNN 的下載網站：

https://developer.nvidia.com/rdp/cudnn-download

值得一提的是，下載之前是需要先注冊賬號。在下圖所示的版本列表中，選擇適配 CUDA9.0 的，對應的是 Linux 系統。

2、下載完畢之后，解壓：

$ tar -zxvf cudnn-9.0-linux-x64-v7.6.1.34.tgz

解壓命令輸出如下：

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

拷貝相關文件至指定目錄：

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

3、到本地 cuda 路徑，創建軟連接，終端輸入：

$ 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

出現 cuDNN 對應的版本號就證明安裝成功！

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 顯卡驅動 + CUDA9.0 + cuDNN7 環境配置完成。

參考文獻：

https://blog.csdn.net/weixin_41864878/article/details/79621210

---

推薦閱讀

（點擊標題可跳轉閱讀）

完備的 AI 學習路線，最詳細的資源整理！

干貨 | 公眾號歷史文章精選

我的深度學習入門路線

我的機器學習入門路線圖

覺得這篇文章有幫助？請轉發給更多人

關注?AI有道?加星標，獲取最新 AI 干貨

最新 AI 干貨，我在看??

總結

以上是生活随笔為你收集整理的手把手教你在 Ubuntu16.04 安装 GPU 驱动 + CUDA9.0 + cuDNN7的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。