Ubuntu16.04 / Windows10 安装 Tensorflow GPU版教程 CUDA9.0 - 请转CUDA8.0

最新更新:2018-03-16 16:07:26

注意这几点,少填坑:

安装顺序: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,到这一步基本就OK了。

一定要下载对应的CUDA和CUDNN。

Tensorflow不支持CUDA9.0,故请将以下所有涉及9.0的换成8.0即可。

 

 


 

下面的内容仅供参考。

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,否则白下。 

 

最后放一张

 

posted @ 2017-11-07 09:28  小鑫同学  阅读(7945)  评论(0编辑  收藏  举报

© 2018 追梦网 | Created by Cai M$ | 博客园 | 推荐使用 Microsoft EdgeWindows 10