1. 程式人生 > >Tensorflow+cuda+cudnn+window+Python之window下安裝TensorFlow

Tensorflow+cuda+cudnn+window+Python之window下安裝TensorFlow

這篇文章主要講TensorFlow在原生Windows下的安裝教程。

安裝環境:TensorFlow0.12+cuda8.0+cudnn v5.1+window7+python3.5

(1)  先安裝Python3.5

從官網下載Python3.5,https://www.python.org/downloads/windows/,雙擊安裝即可。接著將安裝路徑新增進環境變數中。具體步驟:計算機右鍵屬性—高階系統設定—環境變數—系統變數—找到Path,把Python安裝路徑新增進去。

(2)安裝Cuda和Cudnn(CPU版本的不需要安裝,可直接跳過第二步)

首先要檢查電腦的顯示卡型號是否支援GPU加速。

下載完直接雙擊安裝即可。

Cudnn解壓後將bin,include,lib三個資料夾裡面的內容覆蓋至Cuda安裝目錄下,預設路徑為C:\Program Files\NVIDIA GPUComputing Toolkit\CUDA\v8.0(記住不是替換,是把Cudnn檔案裡的.dll檔案新增到Cuda裡面)

接下來,檢驗Cuda是否安裝成功:

進入cuda路徑,預設C:\ProgramData\NVIDIACorporation\CUDA Samples\v8.0\bin\win64\Release執行裡面的deviceQuery.exe檔案,如下圖所示:


                                                                   若出現上圖結果,即說明你安裝cuda成功。

(3)安裝tensorflow

在安裝TensorFlow之前,先修改映象

國外映象使用pip安裝速度很慢,為了使用國內映象加速pip安裝,需要如下修改:

WIndows 7“C:\Users\使用者名稱\AppData\Local\pip”資料夾下,新建文字檔案,新增內容:

1.  [global]  

2.  index-url = http://mirrors.aliyun.com/pypi/simple/  

3.  [install]  

4.  trusted-host=mirrors.aliyun.com  

ps:這裡使用的是阿里雲的國內映象。

接著,在開始選單開啟windows PowerShell,輸入py -3 –m pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl  

Ps:因為本人的電腦同時裝了Python2.7和Python3.5,所以必須指明是使用哪一個版本的pip模組,若讀者電腦只裝了Python3.5,那麼前面的py -3 –m語句可以忽略。

安裝完成後開啟Python 3.5 64bit 進行測試。

所有程式下找到"Python 3.5 64bit",出現命令視窗,輸入測試程式碼:

1.  >>>import tensorflow as tf  

2.  >>>sess = tf.Session()  

3.  >>>x = tf.constant(5)  

4.  >>>y = tf.constant(35)  

5.  >>>print(sess.run(x * y))  

6.  175

測試成功,TensorFlow安裝完成。

最後,檢驗一下cudnn是否配置成功:

Windows PowerShell 下輸入python(因為我的電腦同時裝了Python2.7和3.5,所以必須用py -3指定用哪個版本,若讀者只裝了Python3.5,那麼直接輸入python即可),再輸入import tensorflow,若出現下圖所示,不提示沒有安裝cudnn,則說明cudnn配置成功。


以上,便是本人在安裝TensorFlow時的具體流程,如有差錯,還望不吝賜教!

參考: