本文参考了
http://www.cnblogs.com/5211314jackrose/p/7081020.html
https://jingyan.baidu.com/article/4853e1e55679491909f726f4.html
http://www.cnblogs.com/iloveblog/p/7683349.html
1. 检查自己的GPU是否是CUDA-capable
xtu@xtu-Precision-Tower-7910:~$ lspci | grep -i nvidia 03:00.0 VGA compatible controller: NVIDIA Corporation Device 1b00 (rev a1) 03:00.1 Audio device: NVIDIA Corporation Device 10ef (rev a1)
下载NVIDIA-Linux-x86_64-384.90.run
网址http://www.geforce.cn/drivers/results/123918
下载cuda_9.0.103_384.59_linux.run,注意要下载runfile文件
网址https://developer.nvidia.com/cuda-downloads
下载cudnnv7.0,我下载地时候只有一个版本支持cuda9.0
3.安装nvidia驱动
这里遇到了很多坑,有的比较独特,我在网上很难查到
3.1 禁用nouveau
ubuntu自带的nouveau驱动会影响cuda安装,不当操作会导致黑屏和登陆循环。 终端运行:
lsmod | grep nouveau如果有输出则代表nouveau正在加载。则需要禁用nouveau,在
/etc/modprobe.d
中创建文件
blacklist-nouveau.conf,再用getid打开
cd /etc/modprobe.d
/etc/modprobe.d$ sudo touch blacklist-nouveau.confsudo gedit blacklist-nouveau.conf
在文件中输入以下内容并保存:
blacklist nouveau options nouveau modeset=0
之后更新
sudo update-initramfs -u
lsmod | grep nouveau
这种方式也可能不能彻底禁用nouveau,在此基础上可以移除以下文件:nouveau.ko;