Ubuntu CUDA+cudnn+驱动安装(被坑过无数次的小白的经验)

PS: 这仅仅是其中一种可行的方法 没成功安装过的可以看看

tensorflow的要求版本可能会变化，一定要按照官网上建议的装GPU support

安装Ubuntu

最新版本是18.04 但cuda那边并不支持18 所以我安装的是16.04版本 安装可以参考Ubuntu双系统安装

安装CUDA

一定要安装 9.0 的 下载地址
然后选择 deb的network版本 下载好之后复制到 home 主目录下
之后执行下面的命令

第4步操作要改为 这样才能安装9.0的

sudo apt-get install cuda-9.0

然后中途可能会出现一个对话框 我选的是YES 接着随便输个密码就行 其他默认yes就行

然后配置环境变量

gedit ~/.bashrc

加入

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/lib/nvidia-390"
export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export PATH=/usr/lib/nvidia-390/bin${PATH:+:${PATH}}

其中版本号（也就是数字的地方）改一下

刷新环境变量

source ~/.bashrc

测试一下nvcc

nvcc -V

没蹦出提示错误提示就是成功啦

接着试试跑样例

cd cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

如果跳出你的显卡信息就是成功啦 ， 如果失败了就接着看下去

安装驱动

这一步仅仅针对nvcc-V成功 但 样例失败的情况(就是这个浪费了我两天)
首先 去英伟达官网下载对应的驱动下载地址 tensorflow对驱动有版本要求，所以要先去这里看看版本要求

禁用secure boot
在BIOS中将secure boot设置disable

禁用nouveau第三方驱动

sudo gedit /etc/modprobe.d/blacklist.conf

在最后添加

blacklist nouveau

更新内核

sudo update-initramfs -u

重启后 输入

lsmod | grep nouveau

没有输出就大功告成啦
按Ctrl + Alt + F2 重新输入你的用户名 密码 (最好先拍张照哦，按了之后就看不见你的浏览器了)

接着按执行下面代码

service lightdm stop
sudo chmod a+x NVIDIA-Linux-x86_64-xxx.xx.run
sudo ./NVIDIA-Linux-x86_64-xxx.xx.run –no-opengl-files

会有几执政个选项 直接默认就行了

等他执行完了之后重启就OK 不会重启的直接按电源
然后执行上面出错的地方 就可以啦(反正我是可以的)

安装CUDNN

下载地址
这里要下载那个cuda9.0的linux文件 就是下面这个


下载下来会是一个tgz文件
解压他到local文件夹下

tar -zvxf 文件名.tgz -C /usr/local

sudo chmod a+r /usr/local/cuda/include/cudnn.h  

执行完之后

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

有输出就是成功安装cudnn了


至此cuda和cudnn就安装成功了


有不对的地方欢迎大佬指正

参考：
Linux安装NVIDIA显卡驱动的正确姿势
最正确的姿势安装cudnn，网上大多数教程都太坑了