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*
原文地址:https://www.cnblogs.com/code-wangjun/p/9986255.html