深度学习环境配置,安装Nvidia显卡驱动,CUDA以及cudnn
OS:ubuntu 16.04;
driver: nvidia 396;
CUDA: 9.2
cudnn: 9.2
卸载原有Nvidia驱动
1 | # 卸载原有Nvidia驱动 |
安装驱动
apt-get安装而非下载安装,有一些博客说下载安装总有问题
然而,我两种方法都没装上(见处理 nouveau)
1 | # apt-get安装 |
如果显卡驱动装不上,先处理 nouveau
1 | lsmod | grep nouveau |
如果有输出则代表nouveau正在加载。则需要禁用nouveau
/etc/modprobe.d中创建文件blacklist-nouveau.conf,再用gedit打开
1 | cd /etc/modprobe.d |
在文件中输入以下内容并保存:
1 | blacklist nouveau |
之后更新
1 | sudo update-initramfs -u |
再次查看
1 | lsmod | grep nouveau |
这种方式也可能不能彻底禁用nouveau,在此基础上可以移除以下文件:nouveau.ko;nouveau.ko.org
1 | cd /lib/modules/4.4.0-83-generic/kernel/drivers/gpu/drm/nouveau |
再更新
1 | 大专栏 Gnome Ubuntu16安装Nvidia显卡396驱动,CUDA9.2以及cudnn9.2e">sudo update-initramfs -u |
时重启,再用终端检测一下
1 | lsmod | grep nouveau |
没有输出即为禁用成功。
其实上面这些操作我做完之后,重启还是黑屏了;后来重新在命令行中删除Nvidia的东西再装驱动再重启才好的
nvidia驱动安装成功界面
下载文件
执行
1 | Installation Instructions: |
看网上有人用deb安装会有问题,但是我这么装没事儿呀
检测CUDA是否安装好
1 | cd /usr/local/cuda-9.2/samples/1_Utilities/deviceQuery |
如果显示的是关于GPU的信息,则说明安装成功了。
别人用runfile安装的时候 有几个步骤需要选择,其中有个选择是否安装nvidia驱动的要选N!我之前就在这里装错了,超烦
还有有一些教程说要在~/.bashrc里面增加环境变量
1 | export PATH=/usr/local/cuda-9.2/bin${PATH:+:${PATH}} |
cuDNN安装
1 | tar -zxvf cudnn-9.2-linux-x64-v7.2.1.38.tgz |
然后可以执行
1 | nvcc -V |
查看是否成功