本文記錄搭建的基本步驟及遇到的問題。
0 換國(guó)內(nèi)源+顯卡驅(qū)動(dòng)更新
更換Ubuntu下載源
參考鏈接:Ubuntu圖形界面換源
顯卡驅(qū)動(dòng)更新
法一:通過圖形界面
應(yīng)用程序-->軟件與更新-->開發(fā)者選項(xiàng)-->勾選提前釋放出的更新
軟件與更新-->附加驅(qū)動(dòng)-->選擇nvidia最新驅(qū)動(dòng)混聊,然后選擇應(yīng)用更改
法二:利用終端命令行
打開終端坝咐,按照如下步驟(不用輸入每行代碼前的$),即可更新英偉達(dá)驅(qū)動(dòng)到指定版本。
$ sudo apt update
$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C03sv00001043sd000085ABbc03sc00i00
vendor : NVIDIA Corporation
model : GP106 [GeForce GTX 1060 6GB]
driver : nvidia-driver-410 - third-party free
driver : nvidia-driver-430 - third-party free recommended
driver : nvidia-driver-396 - third-party free
driver : xserver-xorg-video-nouveau - distro free builtin
# 或者安裝指定驅(qū)動(dòng)(這里挑選自己需要的驅(qū)動(dòng),然后安裝即可,不必卸載之前裝過的驅(qū)動(dòng))
$ sudo apt install nvidia-driver-430
1 cuda10的安裝
1.1 下載對(duì)應(yīng)版本的安裝包
CUDA Toolkit | Linux x86_64 Driver Version | Windows x86_64 Driver Version |
---|---|---|
CUDA 10.1.105 | >= 418.39 | >= 418.96 |
CUDA 10.0.130 | >= 410.48 | >= 411.31 |
CUDA 9.2 (9.2.88) | >= 396.26 | >= 397.44 |
CUDA 9.1 (9.1.85) | >= 390.46 | >= 391.29 |
CUDA 9.0 (9.0.76) | >= 384.81 | >= 385.54 |
CUDA 8.0 (8.0.61 GA2) | >= 375.26 | >= 376.51 |
CUDA 8.0 (8.0.44) | >= 367.48 | >= 369.30 |
CUDA官方下載鏈接:https://developer.nvidia.com/cuda-10.0-download-archive
1.2 安裝CUDA
cp Dowloads #跳轉(zhuǎn)到下載的安裝包所在目錄瞄沙,若已在該目錄为朋,則無需跳轉(zhuǎn)
sudo chmod 777 cuda_10.0.130_410.48_linux.run
sudo ./cuda_10.0.130_410.48_linux.run
先按q退出說明(more...),然后安裝設(shè)置如下(一定要注意看下面的說明):
- 驅(qū)動(dòng)在之前單獨(dú)安裝传蹈,所以這里NVIDIA Accelerated Graphics Driver選n(避免安裝報(bào)錯(cuò))押逼;
- 關(guān)于是否創(chuàng)建軟鏈接symbolic link:如果后續(xù)需要安裝多個(gè)CUDA版本,這里可以選擇n惦界,但后面一定要記得手動(dòng)創(chuàng)建軟鏈接(即必須執(zhí)行步驟1.4)挑格,否則cudnn的示例程序會(huì)報(bào)錯(cuò),如果只裝一個(gè)CUDA版本沾歪,就可以選擇y(不需要執(zhí)行步驟1.4)漂彤;
1.3 將cuda、cuda10.0都配置到環(huán)境變量
如果忘記配置環(huán)境變量,可能無法找到nvcc等命令挫望,很重要立润。
打開終端,在終端中執(zhí)行如下語(yǔ)句
sudo gedit ~/.bashrc
此時(shí)會(huì)有一個(gè)文本被打開媳板,將文本最小化桑腮,繼續(xù)在文本末尾中加上如下幾行代碼:
export PATH=$PATH:/usr/local/cuda
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH
保存之后即可關(guān)閉文本,然后再終端中運(yùn)行如下代碼蛉幸,使其生效破讨。
source ~/.bashrc
參考博文鏈接:https://blog.csdn.net/qq_31456593/article/details/90170708
1.4 切換CUDA版本(在是否安裝symbolic link時(shí)選了n的話,需要執(zhí)行此步驟奕纫,如果選的y則無需執(zhí)行此步驟)
sudo rm -rf cuda #刪除現(xiàn)有的軟鏈接
sudo ln -s /usr/local/cuda10.0 /usr/local/cuda #建立新的軟鏈接
nvcc --version #查看當(dāng)前使用的cuda版本
在下面這個(gè)圖中提陶,帶箭頭的cuda文件是軟鏈接,而cuda10.0是咱們剛剛安裝的版本匹层,我們將cuda這個(gè)軟鏈接指向cuda10.0搁骑,即可完成版本切換。先跳轉(zhuǎn)到cuda文件夾所在目錄又固,一般在/usr/local里面仲器。利用stat cuda可以驗(yàn)證cuda環(huán)境變量是否添加成功,nvcc -V可以驗(yàn)證cuda10.0的環(huán)境變量是否添加成功:
stat cuda
nvcc --version
運(yùn)行結(jié)果如上圖所示仰冠,可以看到cuda軟鏈接指向了cuda10.0乏冀,當(dāng)前運(yùn)行的是cuda10.0的版本。
本節(jié)參考1:http://www.reibang.com/p/6a6fbce9073f
本節(jié)參考2:https://blog.csdn.net/lxx4610/article/details/105795408
2 cuDNN7.6.2安裝
這里我選用了第二種下載方法:下載了3個(gè)deb包來安裝洋只,見下圖最底部的三個(gè)鏈接
cuDNN官方下載鏈接:https://developer.nvidia.com/rdp/cudnn-download(需要登陸辆沦、并勾選同意協(xié)議后,才能下載)
下圖中2.3.2是通過Deb安裝包安裝识虚,2.4小結(jié)是驗(yàn)證cuDNN是否安裝成功肢扯。圖片來自cuDNN:官網(wǎng)的安裝教程
易錯(cuò)提醒:記得要將cuda和cuda10.0分別添加到環(huán)境變量担锤!且需要將cuda鏈接到安裝好的cuda10.0蔚晨!否則CUDNN驗(yàn)證時(shí)會(huì)各種報(bào)錯(cuò)(比如不存在這個(gè)文件或目錄等。)另外肛循,cuDNN官網(wǎng)上下載的三個(gè)deb包容易出錯(cuò)铭腕,如果cuDNN安裝時(shí)報(bào)錯(cuò),可以重新下載多糠。