點(diǎn)擊上方“AI有道”，選擇“星標(biāo)”公眾號(hào)

重磅干貨，第一時(shí)間送達(dá)

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

注意：由于不同系統(tǒng)，不同 GPU 對(duì)應(yīng)的 CUDA 和 cuDNN 均有差異，本文僅以 Ubuntu16.04、NVIDIA GeForce GPU、CUDA9.0 和 cuDNN7 為例，來(lái)介紹安裝過(guò)程。其它版本過(guò)程類(lèi)似，但存在細(xì)微差別，請(qǐng)讀者以實(shí)踐為主。

必要條件

工欲善其事必先利其器，首先確定你的電腦滿(mǎn)足下面兩個(gè)條件。

• 安裝 Ubuntu16.04

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

一、安裝 GPU 顯卡驅(qū)動(dòng)

有一塊 NVIDIA GPU 顯卡 還不夠，還需要安裝 GPU 顯卡驅(qū)動(dòng)。

1、首先登陸 GeForce 官網(wǎng)：

https://www.geforce.cn/drivers

在手動(dòng)搜索驅(qū)動(dòng)程序里，選擇自己對(duì)于的系統(tǒng)，注意是“Linux 64-bit”。其他選項(xiàng)默認(rèn)即可。然后，點(diǎn)擊開(kāi)始搜索，將會(huì)自動(dòng)搜索你的電腦相匹配的 GPU 顯卡驅(qū)動(dòng)，搜索結(jié)果如下圖所示：

如上圖所示，結(jié)果會(huì)顯示可用的 GPU 顯卡驅(qū)動(dòng)程序，確定你需要的版本，直接點(diǎn)擊就可以下載了。例如我選擇的版本是418.43，下載后的文件是 NVIDIA-Linux-x86_64-418.43.run。

2、打開(kāi)終端 Terminal，首先卸載系統(tǒng)之前安裝的 NVIDIA 顯卡驅(qū)動(dòng)，以防萬(wàn)一。（沒(méi)有安裝的可以忽略這一步）

$ sudo apt-get --purge remove nvidia-*

3、打開(kāi) blacklist：

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

在最后添加幾行：

blacklist vga16fb blacklist nouveau blacklist rivafb blacklist nvidiafb blacklist rivatv

這時(shí)候在終端輸入：

$ lsmod | grep nouveau

如果沒(méi)有輸出，可以直接進(jìn)行下一步，安裝驅(qū)動(dòng)。如果仍有輸出，則在終端輸入：

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

直接寫(xiě)入：

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

保存關(guān)閉，再在終端輸入下面的命令驗(yàn)證是否沒(méi)有輸出了。

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

到這一步，應(yīng)該沒(méi)有輸出了，可以進(jìn)行安裝驅(qū)動(dòng)了。

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

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

進(jìn)入 .run 文件的下載路徑，使用 chmod 命令增加 .run 文件的運(yùn)行權(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文件

然后就會(huì)開(kāi)始安裝程序，安裝完成之后，輸入下面的命令：

$ nvidia-smi

如果出現(xiàn)與下面類(lèi)似的結(jié)果，證明驅(qū)動(dòng)安裝成功了。

最后，回到圖像界面。

$ sudo /etc/init.d/lightdm start

這樣，驅(qū)動(dòng)就算正式安裝成功了。

二、安裝 CUDA9.0

1、首先，登錄 CUDA 的下載網(wǎng)站：

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

由于現(xiàn)在 CUDA 的版本已經(jīng)更新到 10 了，我們需要安裝 9.0 版本。因此，點(diǎn)擊 Legacy Releases，進(jìn)入之前的版本列表。

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

選擇對(duì)應(yīng)的系統(tǒng)等信息，就可以點(diǎn)擊 Download 進(jìn)行下載了。注意選擇下載 runfile 文件。

2、打開(kāi)終端，進(jìn)入 CUDA9.0 的下載目錄，輸入：

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

然后就會(huì)進(jìn)入安裝。在安裝過(guò)程中會(huì)詢(xún)問(wèn)你一些配置信息，一般直接選擇默認(rèn)選項(xiàng)就行。有一個(gè)必須注意的地方是會(huì)讓你選擇是否安裝 Graphics Driver for Linux-x86_64：XXXX 版本，這個(gè)地方必須選擇 no！否則會(huì)覆蓋之前的驅(qū)動(dòng)。

3、安裝完畢之后，聲明環(huán)境變量。打開(kāi) .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、驗(yàn)證 GPU 驅(qū)動(dòng)版本：418.43

$ cat /proc/driver/nvidia/version

驗(yàn)證 CUDA 版本：9.0

$$ nvcc -V

5、還有可能遇到的問(wèn)題是沒(méi)有安裝 gcc 或者 gcc 版本過(guò)高需要手動(dòng)進(jìn)行降級(jí)，都比較好解決，這里就不再贅述了。

三、安裝 cuDNN

1、首先，進(jìn)入 cuDNN 的下載網(wǎng)站：

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

值得一提的是，下載之前是需要先注冊(cè)賬號(hào)。在下圖所示的版本列表中，選擇適配 CUDA9.0 的，對(duì)應(yīng)的是 Linux 系統(tǒng)。

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

拷貝相關(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

注意，這里的文件名即版本號(hào)就參照之前解壓出來(lái)的文件名對(duì)照著寫(xiě)。

4、到本地 cuda 路徑下檢測(cè)一下：

$ cd /usr/local/cuda/lib64 $ ll

出現(xiàn) cuDNN 對(duì)應(yīng)的版本號(hào)就證明安裝成功！

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ū)動(dòng) + CUDA9.0 + cuDNN7 環(huán)境配置完成。

參考文獻(xiàn)：

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

---

推薦閱讀

（點(diǎn)擊標(biāo)題可跳轉(zhuǎn)閱讀）

完備的 AI 學(xué)習(xí)路線，最詳細(xì)的資源整理！

干貨 | 公眾號(hào)歷史文章精選

我的深度學(xué)習(xí)入門(mén)路線

我的機(jī)器學(xué)習(xí)入門(mén)路線圖

覺(jué)得這篇文章有幫助？請(qǐng)轉(zhuǎn)發(fā)給更多人

關(guān)注?AI有道?加星標(biāo)，獲取最新 AI 干貨

最新 AI 干貨，我在看??

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。