Linux從入門到適應(四):Ubuntu 16.04環境下,安裝Nvidia驅動,cuda9.2和 cudnn

在安裝深度學習框架以前,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*
相關文章
相關標籤/搜索