Ubuntu16.04 / Windows10 安装 Tensorflow GPU版教程 CUDA9.0

yzwj2004

　　Last upd: 10:45:48 2017-11-08
　　注意这几点，少填坑：
　　安装顺序：Python -> Tensorflow -> Nvidia GPU Driver -> CUDA -> CUDNN
　　其中，可以在安装完成TensorFlow之后，尝试在python中import tensorflow，然后看给出的错误信息，其中包含所需要的CUDA和CUDNN版本，比如：



ImportError: libcudnn.so.6: cannot open shared object file: No such file or directory
　　表示需要CUDNN6.0版本，而与其对应的是CUDA8.0。
　　一定要下载对应的CUDA和CUDNN。
　　Tensorflow不支持CUDA9.0，故请将以下所有涉及9.0的换成8.0即可。
　　下面的内容可以不用看了，仅供参考。
　　建议看这一篇：http://www.cnblogs.com/xuliangxing/p/7575586.html
　　=====================
　　Windows10中，下载CUDA 8.0之后，在安装的时候选择自定义安装，并去掉GEFORCE和驱动，否则会安装失败。
　　安装完成之后，在python中调用tensorflow发现缺少cudnn6_64.dll，这意味着需要下载cudnn6 for cuda8.0。
　　下载完了直接将文件放到对应的文件夹中即可。
　　再次调用tensorflow，成功。
　　=====================
　　最终成功：
　　CUDA和CUDNN的测试也通过了，就不放图了。放一张import tensorflow成功的。
　　

　　==============================
　　1、下载CUDA 9.0
作者：小鑫同学，原文链接
　　从官网下载CUDA9.0，注意，如果网速太慢，不能10分钟之内下载完毕，最好用虚拟主机或者百度网盘离线下载。
　　我用本机下载尝试了好几次，都失败了，最后用的是Azure上的Ubuntu下载的，接着用/var/www/html分享，此外还有NFS、SSH、FTP等方法，可以在本地共享远程的ubuntu文件。Azure上的磁盘是托管的，如果删除了虚拟主机，应该可以直接全部导出磁盘上的文件，不过我没有试过。
　　Linux上的常用下载软件是uget，用uget代理firefox的全部下载链接。可以先用uget从官网下载，失败之后千万不要重试，让其hang out，然后再从虚拟主机或者其他的链接用uget下载，这时uget会自动续传。
　　2、安装NVIDIA显卡驱动
作者：小鑫同学，原文链接
　　这个时候，直接在tty1中操作，请确保有网络链接。



sudo service lightdm stop #关闭图形界面
　　黑屏、开机时候的样子，接着按 ctrl+alt+F1进入tty1，登录。
　　我们发现下载的CUDA文件名中有384字样，这意味着我们需要下载nvidia-384这个驱动。为了避免出现循环登录、“Low graphical mode”以及不识别各种硬件的问题，我们需要先将默认的若干显卡驱动添加到一个配置文件中：



sudo apt-get install vim  #自带的vi是屎
cd /etc/modprobe.d/
sudo vim blacklist.conf

作者：小鑫同学，原文链接
　　并且在最后添加：blacklist nouveau



sudo update-initramfs -u

　　接着直接安装nvidia-384



sudo apt-get remove --purge nvidia-* #先卸载已经安装的驱动
sudo apt-get install nvidia-384 nvidia-xconfig nvidia-settings nvidia-prime　
sudo update-initramfs -u
　　3、安装CUDA9.0
作者：小鑫同学，原文链接
　　依然实在tty1中进行，找到下载的CUDA文件，参照官网在下载按钮下方给出的安装方法，输入命令即可。我的是.run文件，因此



sudo sh cuda_9.0_file.run
　　按空格快速浏览协议，accept之后，不用安装cuda中带的384驱动，输入n。接着输入y或者按回车。
　　安装完毕之后，重启。



sudo reboot
　　至此，CUDA安装完毕。新安装的显卡驱动对桌面的支持不好，在拖动窗口的时候会有锯齿。╮(╯▽╰)╭（我自己安装的时候都是坑。）
作者：小鑫同学，原文链接
　　安装Tensorflow-gpu



sudo pip3 install tensorflow-gpu
　　一定要用sudo，否则白下。
　　最后放一张