在安装深度学习框架之前,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*