这个东西搞了好久,还好比较有死磕的精神终于倒腾好了。。
PS:其实也很简单,自己被坑的关键原因是自己盲目观看各个博客教程,而tensorflow的配置一直在更新,而我没有注意阅读官方的文档,导致cuda和cudnn等一系列不匹配。
一共以下几个部件:
cuda(检查显卡是否有此模块)
cudnn
python
anaconda(之前看很多教程都有这个部分,我就跟着装了,但不知道不装这个能不能完成)
(对了还有我是win10)
下面阐述关键操作:
pip install tensorflow-gpu(后面发现是1.3版本(而很多教程里版本不一致))
而查看文档后发现最新tensorflow支持python3.6(很多之前的教程都采用的python3.5)
安装cuda(nvidia官网好像只提供最新cuda版本(我被提供了9.0),这就是导致我出错(#issue)的关键因素,查看官方文档后才发现tensorflow1.3只支持cuda8.0(至于cuda8.0的文件我是在别人教程里提供的链接得到的,而里面打包提供的cudnn版本为5.1))
接着安装cudnn(需要到官网里找与cuda匹配的版本,这里与cuda8.0匹配两个版本的cudnn(5.1和6,而很多的教程是5.1),而根据官方文档,已经不支持5.1版本,我们选择6)
将下载的cudnn(下载下来的文件名为cuda)的bin路径添加到Path里。
完成上述操作应该可以正常import tensorflow了(我是这样)
我相信大多数人都在这里被坑了,所以综上,我的配置以后可能也不合适了,关键还是要学会看官方文档!!!