為了在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.

然而點了”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’

最後看到一個人說, cuDNN v5才可以, cuDNN v6 不支援, 我就下了cuDNN v5.1, 然後把環境變數PATH改成了D:\NVIDIA\CuDnn_5.1\bin, 就行了… 行了… 了… ….