Windows環(huán)境下搭建TensorFlow(GPU)簡單教程

一、環(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

一定要下載圖中圈出的版本,不然會(huì)死的很慘

將下載好的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,問題解決!

此處參考http://blog.csdn.net/fool_frog/article/details/53422460


六、可能需要的組件

image (1.5.5)//圖像處理相關(guān)

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容