Ubuntu 16.04環境下,安裝Nvidia驅動,cuda9.2和 cudnn

原文鏈接:https://www.cnblogs.com/code-wangjun/p/9986255.html

在安裝深度學習框架之前,cuda和cudnn是必須要提前安裝的,現在按照流程而nvidia驅動的版本和cuda版本有這一些對應關係,所以需要按照版本進行安裝,現在說一下如何安裝:

1 安裝nvidia驅動:

(1)卸載舊版本的顯卡驅動:

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/。

(4)禁用nouveau第三方驅動,

        打開配置文件:sudo gedit /etc/modprobe.d/blacklist.conf

        在最後一行增加 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 中設置環境變量:

1

2

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文件夾下對應的位置:

1

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)修改文件權限:

1

sudo chmod a+r /usr/local/cuda-9.2/include/cudnn.h /usr/local/cuda-9.2/lib64/libcudnn*

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章