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


在安裝深度學習框架之前,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 中設置環境變量:

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*


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM