WIN10 + python 3.5 + VS2017
装之前上网稍微查一查发现好像非常多的坑,各种的错误。自己装的时候也出错了一下,然而仔细看看会发现,出错的都是旧版本,如果是从零开始安装新的tensorflow的话是不会有任何问题的。所以这个也不叫什么教程就是一点提示而已233
由于我本身就安装了VS2017,所以并不确定没有安装VS行不行,理论上好像是可以的。
1、安装Tensorflow-gpu
pip3 install tensorflow-gpu
cmd下用pip安装很简洁。唯一会出问题的下载本身的问题,网络有时候不行的话下载到一半就会报错,如果出错的话建议换一个pip的下载源,或者关了重新下试试看(我就是下了三遍才下好的结果大佬告诉我换下载源试试)
2、安装CUDA9.0
9.0!一定是9.0!tensorflow每个版本对应的cuda都不一样,按照我当前的日期。18年2月25日,tensorflow1.5版本,就是对应CUDA9.0,如果以后更新了新的版本,就要自己去百度一下,看看官方文档写明的对应的版本是什么。或者你直接import tensorflow的话,也会发现报错,报错提示里面也会说缺少哪个版本的模块的。
百度直接进官网下载就可以了,注意选历史版本找9.0
3、安装cuDNN7
强调版本:7.0,原因如上了。这个下载之前要注册,没有什么要求的,之前注册登陆下载就可以了。然后把下载到的东西放到对应的安装目录里就可以了
然后就搞定了,tensorflow新版本安装已经很方便了,大多数错误都是旧版本的问题233。