tensorflow1.5.0,cuda9,cudnn7安装

最近看到消息,Googlle的tensorflow1.5发布了——

更新后的tensorflow最重要的points对我来说是支持cuda9和cudnn7,具体还有其他的改变见github吧:

https://github.com/tensorflow/tensorflow/releases

或者看这篇文章:https://news.cnblogs.com/n/586539/


很多功能没用过,没接触过,没懂过(哈哈哈哈哈~~~),下面开始安装的教程吧:

Anaconda对应Python版本:python3.6.3

Cuda:9.0

cudnn:7

tensorflow版本:1.5


1、使用anaconda自带的pip安装工具进行安装,输入命令:

pip install tensorflow-gpu==1.5rc1 --user



2、在输入下面命令后,遇到了这个问题:

TypeError: parse() got an unexpected keyword argument 'transport_encoding'


这个问题搜到两个解决方案:

解决方案1:http://blog.csdn.net/cheese_pop/article/details/78201359,这个方案是安装html5lib来解决解析出现的问题。(我使用了该教程没能解决,这个教程的更新后也加上了解决方案2,我开始没注意,解决方案2是在stackoverflow上找的,下面给stackoverflow的链)

解决方案2:https://stackoverflow.com/questions/46499808/pip-throws-typeerror-parse-got-an-unexpected-keyword-argument-transport-enco?rq=1,这个方案是使用conda更新pip

tensorflow1.5.0,cuda9,cudnn7安装_第1张图片

两个解决方案都尝试了下,解决方案1没能解决,执行解决方案2后,可以输入步骤1中的命令进行安装了。


在进行解决方案1时,发现自己安装的anaconda的conda版本低了:

tensorflow1.5.0,cuda9,cudnn7安装_第2张图片

使用conda update-n base conda命令更新:(conda自己能更新自己,一直觉得这是个很神奇的事情^-^)

conda update-n base conda
tensorflow1.5.0,cuda9,cudnn7安装_第3张图片


3、更新完conda,pip工具,并且安装好html5lib后,按照步骤1安装tensorflow:

tensorflow1.5.0,cuda9,cudnn7安装_第4张图片


4、安装完tensorflow并不意味着能使用了,我之前装的cuda是9.1最新版,tensorflow支持了9,但是不是9.1,在导入tensorflow的时候,就又遇到了下面的问题,错误提示说的很清楚,找不到cudat64_90.dll:

tensorflow1.5.0,cuda9,cudnn7安装_第5张图片

这个文件在cuda的目录下,在安装cuda时有cuda的环境变量,找到环境变量所在目录(这里已经是我安装完毕9.0后的,9.1的也在这个位置),复制环境变量的值:

tensorflow1.5.0,cuda9,cudnn7安装_第6张图片

将上述复制的内容复制到文件管理的地址栏中,进入该目录后,可以看到cuda的对应目录,并点开bin目录(目录文件路径没有抹掉,暴露就暴露了吧~):

tensorflow1.5.0,cuda9,cudnn7安装_第7张图片


可以在bin目录下面看到上面缺失的文件,之前安装的是9.1,对应的dll为cudat64_90.dll(我之前简单粗暴地把名字给改了,好像可以,不过后来还是重装了cuda9.0),cuda的安装可以参考我的另一篇博文:http://blog.csdn.net/zhangwellyear/article/details/78992661,不同的是cuda的版本,9.0找一下还是可以找到的。

这里给出cuda9.0的下载链接:https://developer.nvidia.com/cuda-90-download-archive


5、安装cuda完毕后,再次导入tensorflow,还会遇到错误,错误提示为缺失了cudnn64_7.dll文件:

tensorflow1.5.0,cuda9,cudnn7安装_第8张图片

该文件不需要安装,下载即可,下载链接为:https://developer.nvidia.com/rdp/cudnn-download

下载时需要进行注册,填几个东西,怕麻烦的,在百度云这里下吧:https://pan.baidu.com/s/1dFZvOKL  密码:auca


6、上述文件下载后解压,将cudnn64_7.dll添加到环境变量,再次导入tensorflow,并查看tf的版本,跟下面一样的时候,就安装成功了(添加环境变量网上有很多,就不在此赘述了):

tensorflow1.5.0,cuda9,cudnn7安装_第9张图片

7、使用一段代码测试下GPU和CPU,参考的文章地址为:

https://learningtensorflow.com/lesson10/


8、使用《Tensorflow 实战Google深度学习框架》这本书里的迁移学习代码测试下(第6章,6.5.2,p161):

代码可以在这里下载:http://download.csdn.net/download/zhangwellyear/10230481,代码对应的Inception模型及训练数据需要自己下载,代码中的目录需要进行修改后才可以使用,上面的测试已经足矣,这里给出的是一个完整的可应用的代码。

tensorflow1.5.0,cuda9,cudnn7安装_第10张图片


参考文章:

https://wongchen.github.io/2018/01/17/cuda9-cudnn7/

http://aifuns.cc/2017/11/12/win10-cuda8-cudnn6-tensorflow-keras-setting.html

你可能感兴趣的:(Python3,机器学习,Cuda,tensorflow,深度学习,深度学习)