在安裝深度學習框架以前,cuda和cudnn是必需要提早安裝的,如今按照流程而nvidia驅動的版本和cuda版本有這一些對應關係,因此須要按照版本進行安裝,如今說一下如何安裝:linux
1 安裝nvidia驅動:ubuntu
(1)卸載舊版本的顯卡驅動:bash
sudo apt-get purge nvidia*
框架
(2)添加Graphic Drivers PPA學習
sudo add-apt-repository ppa:graphics-drivers/ppa
測試
sudo apt-get update
命令行
(3)去英偉達官網,或者使用
ubuntu-drivers devices查看最佳版本的驅動好,下載驅動。驅動選擇.run文件,而且複製移動到一個方便 cd 的英文文件夾下面,好比:/home/。
code
(4)禁用nouveau第三方驅動,blog
打開配置文件:sudo gedit /etc/modprobe.d/blacklist.confget
在最後一行增長 blacklist nouveau
改好後執行命令:sudo update-initramfs -u
重啓:reboot
(5)按Ctrl+Alt+F1進入命令行界面
執行命令:lsmod | grep nouveau
禁用X服務:sudo /etc/init.d/lightdm stop
進入你存放驅動.run文件的文件夾:cd /home
給驅動run文件賦予可執行權限:sudo chmod a+x NVIDIA-Linux-x86_64-410.78.run
安裝 sudo ./NVIDIA-Linux-x86_64-410.78.run -no-opengl-files 安裝過程當中按照提示直接的默認值直接按
開啓X服務:sudo /etc/init.d/lightdm start
重啓,用nvidia-smi檢查
2 安裝cuda
(1)cuda文件去nvidia官網下載,能夠選擇.run或者.deb文件均可以,版本與你的顯卡要對應,以及深度學習的框架是否支持,不建議下載最新版本的。
(2)按照官網下面的提示命令安裝。
(3)在~/.bashrc 中設置環境變量:
exportPATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} exportLD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
運行 source ~/.bashrc 重啓。
(4)檢測cuda安裝是否成功,若是是默認安裝,則cuda的測試sample在/usr/local/cuda-9.2/samples/1_Utilities/deviceQuery (個人cuda版本爲9.2)
直接運行 makefile文件:sudo make
執行生成的可執行文件:./deviceQuery
如何結果爲 deviceQuery, CUDA Driver = CUDART, CUDA Driver Version = 9.2, CUDA Runtime Version = 9.2, NumDevs = 1 Result = PASS 說明安裝成功
3 安裝cudnn
(1)去官網登陸,下載cudnn文件,選擇與cuda版本對應的cudnn文件,接着選擇 cuDNN Library for Linux
(2)解壓縮: tar -zxvf cudnn-9.2-linux-x64-v7.4.1.5.tgz
(3)將相應的頭文件複製到cuda-9.2文件夾下對應的位置:
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.2/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.2/lib64
(4)修改文件權限:
sudo chmod a+r /usr/local/cuda-9.2/include/cudnn.h /usr/local/cuda-9.2/lib64/libcudnn*