windows上安装tensorflow遇到的关键坑

来源:互联网 发布:小电影下载软件 编辑:程序博客网 时间:2024/05/02 00:57

这个东西搞了好久,还好比较有死磕的精神终于倒腾好了。。


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了(我是这样)

我相信大多数人都在这里被坑了,所以综上,我的配置以后可能也不合适了,关键还是要学会看官方文档!!!



#issue截图

ImportError: DLL load failed: 找不到指定的模块。以及ImportError: No module named '_pywrap_tensorflow_internal'。