tensorflow的要求版本可能会变化,一定要按照官网上建议的装GPU support
最新版本是18.04 但cuda那边并不支持18 所以我安装的是16.04版本 安装可以参考Ubuntu双系统安装
一定要安装 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 不会重启的直接按电源
然后执行上面出错的地方 就可以啦(反正我是可以的)
下载地址
这里要下载那个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,网上大多数教程都太坑了