Windows下安装TensorFlow-gpu

为了在windows上也能看看TensorFlow源码, 我决定也装一下. 在这里记录一下遇到的一些坑.

安装Cuda8.0和CuDnn

直接百度cuda和CuDnn, 下载对应的版本. 这一步按照官网的指导进行就能顺利完成.

安装Python3.5.x

到目前(2017.6.1)为止, 官网提供的TensorFlow 1.1版本只支持Python 3.5.x. 我之前安装的Python3.6执行pip install --upgrade tensorflow-gpu时找不到TensorFlow, 所以只能换成了3.5.3

安装TensorFlow-gpu

我先是执行了官网给的pip install --upgrade tensorflow-gpu, 但是速度太慢了, 而且中间因为网速太慢报错:

pip._vendor.requests.packages.urllib3.exceptions.ReadTimeoutError: HTTPSConnectionPool(host=’pypi.python.org’, port=443): Read timed out.

网速慢应该是因为GFW的问题. 于是我去TensorFlow的github主页直接去找下载链接.

然而点了”Python 3.5 64-bit”之后却提示”HTTP ERROR 404”. 于是我又点”build history”, 看到了tensorflow_gpu-1.1.0rc2-cp35-cp35m-win_amd64.whl. 于是下载, cmd执行pip install E:\\tensorflow_gpu-1.1.0rc2-cp35-cp35m-win_amd64.whl, 进行安装.

在安装过程中, 会自动下载很多依赖. 但是由于默认的源在国外, 速度太慢, 因此需要手动换源. 方法如下:
找到C:\Users\xxx\pip, “xxx”是用户名, 新建一个pip文件夹, 并在里面新建一个pip.ini, 打开输入:

[global]
timeout = 6000
index-url = https://pypi.douban.com/simple
trusted-host = https://pypi.douban.com

这里换成了豆瓣的源. 最终顺利安装好.

测试TensorFlow

在cmd中执行python进入Python, 然后执行

import tensorflow

结果报错:

ImportError: No module named ‘_pywrap_tensorflow_internal’

查到了一个解决办法, 还没试:
ImportError: No module named ‘_pywrap_tensorflow’ Failed to load the native TensorFlow runtime
然后还看了一些方法, 但是都不管用.

最后看到一个人说, cuDNN v5才可以, cuDNN v6 不支持, 我就下了cuDNN v5.1, 然后把环境变量PATH改成了D:\NVIDIA\CuDnn_5.1\bin, 就行了… 行了… 了… ….