一、環(huán)境
我也清楚具體原因,Windows下搭建TensorFlow特別矯情。經(jīng)踩坑實(shí)驗(yàn),將我自己搭建成功的步驟記錄下來,具有一定的泛用性
- Windows 10 64位操作系統(tǒng)
- VC++ 2015(VC++ 2015 運(yùn)行庫)
- CUDA8.0(NVIDIA顯卡)
- cuDNN
- Anaconda(Python 3.5)
二、VC++ 2015
由于我本身就有安裝VS2015,故推斷安裝「VC++ 2015 運(yùn)行庫」即可
三、CUDA8.0與cuDNN
其實(shí)沒有嚴(yán)格的安裝順序,沒有NVIDIA顯卡的話不能安裝CUDA,但TensorFlow仍能用電腦的CPU來計(jì)算。
安裝CUDA8.0,下載地址:https://developer.nvidia.com/cuda-downloads

cuDNN是NVIDIA的深度學(xué)習(xí)SDK,下載地址:https://developer.nvidia.com/cudnn

將下載好的cuDNN.zip解壓,將其中的bin、include、lib中的文件拷貝到CUDA安裝目錄下相應(yīng)的文件夾,CUDA的安裝路徑C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0
其他可運(yùn)行版本供參考,來自https://github.com/tensorflow/tensorflow/issues/9469
- tensorflow 1.2 + cudnn 5.1: works fine
- tensorflow 1.2 + cudnn 6.0: ImportError
- tensorflow 1.3 + cudnn 5.1: works fine
- tensorflow 1.3 + cudnn 6.0: works fine
四、Anaconda(Python 3.5)
據(jù)說Python3.5比較穩(wěn)妥,我沒測試
不要按官方版了,不止是配環(huán)境變量那么簡單。Anaconda集成了很多實(shí)用的庫,環(huán)境變量直接配好了,建議使用
相應(yīng)版本 Anaconda3-4.2.0-Windows-x86_64
安裝好以后,按「win」+「R」輸入cmd,呼出控制臺(tái),在小黑窗里輸入“python”,如果能夠正常顯示python版本說明安裝成功
五、安裝TensorFlow
安裝Anaconda后我們有了更好的控制臺(tái),「Anaconda prompt」
不了解的情況下,不要隨便手賤升級pip
通過pip安裝GPU版的命令
pip3 install --upgrade tensorflow-gpu
安裝完畢后,在Python下輸入命令,沒報(bào)錯(cuò)就是安裝成功
import tensorflow as tf
也可以采用官方的例子進(jìn)行驗(yàn)證
用Anaconda安裝完python3.5后,在安裝tensorflow時(shí)出現(xiàn) “Cannot remove entries from nonexistent file c:\program files\anaconda3\lib\site-packages\easy-install.pth” 的問題。查看原因是因?yàn)閟etuptools版本太低,tensorflow要求29.0.1,當(dāng)前版本為27.2.0,在更新setuptools版本的時(shí)候又找不到easy-install.pth,導(dǎo)致更新失敗
運(yùn)行:pip install --upgrade --ignore-installed setuptools,問題解決!
六、可能需要的組件
image (1.5.5)//圖像處理相關(guān)