Windows安裝TensorFlow

標籤：pip安裝   from   epo   www   sdn   resources   started   blog   終端   

 原生Windows安裝TensorFlow 0.12方法標籤： tensorflowwindows2016-12-04 11:23 37737人閱讀 評論(24) 收藏 舉報 分類：TensorFlow

著作權聲明：本文為博主原創文章，未經博主允許不得轉載。

 

2016年11月29日，TF官方宣布0.12版tensorflow支援原生windows作業系統，不在需要通過Docker進行安裝。作為一個tf初學者，也是windows重度依賴使用者，通過在牆裡牆外各種搜尋，終於找到了一種可行的安裝方法。現予以總結，供同行參考。

 

博主機器配置：

 

[html] view plain copy 

1. OS：Window 7 64bit  
2. CPU：Intel i7-2600K  
3. 記憶體：8G  
4. 顯卡：Nvidia GeForce GTX 560  

（有人推薦使用 Windows PowerShell 代替 CMD，所以下面一、二、三、四步均在Power Shell下執行，“開始”->“附件”->“Windows Power Shell”->“Windows Power Shell”）

一、安裝Python

1、通過Pip在Windows上安裝Python

TensorFlow在Windows上只支援64位Python3.5，可以通過Python 3.5 from python.org 或 Python 3.5 from Anaconda 下載並安裝Python3.5.2（注意選擇正確的作業系統）。

 

2、設定環境變數

上一步安裝完畢後，在PowerShell中輸入pip是找不到該命令的，因此將Python安裝路徑下“%安裝路徑%\Scripts”添加到Path下；再到PowerShell中輸入pip看到若干命令提示，則代表python安裝成功（Python安裝包內建pip）。“開始”->“所有程式”，也可以找到Python終端。

 

二、安裝Cuda和CuDNN（CPU版本TensorFlow請忽略本步驟）

TensorFlow分為CPU版和GPU版，如果你打算安裝GPU版，請先安裝如下兩個驅動：

1、CUDA安裝：https://developer.nvidia.com/cuda-downloads

2、CuDNN安裝：https://developer.nvidia.com/cudnn（要註冊Nvidia使用者，並加入CuDNN開發組，填若干問卷就可以下載了）選擇下載版本時要注意和Cuda版本匹配。解壓後覆蓋至CUDA的安裝目錄下

例如：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\

三、修改Pip國內源

為了使用國內鏡像加速pip安裝，需要如下修改：

WIndows 7 在“C:\Users\使用者名稱\AppData\Local\pip”檔案夾下，建立文字檔，新增內容：

 

[html] view plain copy 

1. [global]  
2. index-url = http://mirrors.aliyun.com/pypi/simple/  
3. [install]  
4. trusted-host=mirrors.aliyun.com  

關於修改pip國內源可參考這裡

四、安裝TensorFlow

安裝CPU版TensorFlow 0.12，Power Shell下輸入：

 

[html] view plain copy 

1. pip install --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-0.12.0rc0-cp35-cp35m-win_amd64.whl  

 

安裝GPU版TensorFlow 0.12，Power Shell下輸入：

 

[html] view plain copy 

1. pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl  

官網查看最新版本下載方式

 

通過pip安裝，會提示錯誤“Http error 404”。估計是無法連網導致的，可以將上述安裝tf的命令https及後面的地址複製，並粘貼到瀏覽器地址欄中手動下載檔案。

然後在Power Shell中輸入下述命令實現本地安裝：

 

[html] view plain copy 

1. pip install F:\DevResources\tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl  

出現錯誤：

 

 

不知什麼原因，無法正常下載numpy 1.11.0。如法炮製，手動下載numpy（119MB）並安裝。Numpy

我選擇下載：numpy?1.11.2+mkl?cp35?cp35m?win_amd64.whl。其中1.11.2是nympy的版本，cp35表示支援python3.5，win_amd64就是支援windows64位作業系統。

下載完畢後先安裝Numpy：

 

[html] view plain copy 

1. pip install F:\DevResources\numpy-1.11.2+mkl-cp35-cp35m-win_amd64.whl  

再安裝TensorFlow（重複上面安裝TensorFlow的步驟），安裝成功。

 

 

五、測試TensorFlow

到“所有程式”下找到"Python 3.5 64bit"，出現命令視窗，輸入測試代碼：

 

[html] view plain copy 

1. >>>import tensorflow as tf  
2. >>>sess = tf.Session()  
3. >>>a = tf.constant(10)  
4. >>>b = tf.constant(22)  
5. >>>print(sess.run(a + b))  
6. 32  

 

正確輸出結果32，安裝完畢。

 

上述方法參考了很多網路資料，也有自己的嘗試，如有不妥的地方還請大家批評指正！

 

頂

8

踩

Windows安裝TensorFlow