1. 程式人生 > >windows下tensorflow安裝(包含cuda cudnn)

windows下tensorflow安裝(包含cuda cudnn)

Tensorflow安裝

CPU版:如果你的電腦沒有NVIDIA的顯示卡的話,你就必須選擇安裝這個版本,不過這個版本的安裝要比GPU版的簡單,官方也推薦先用CPU版的來體驗。

GPU版:TensorFlowGPU上執行要比CPU上快很多,如果你的GPU能夠達到要求就可以選擇安裝GPU版。

GPU版的安裝要求

CUDA® Toolkit 8.0

CUDA® Toolkit 8.0的相關驅動(這個在安裝程式裡會帶)

cuDNN v5.1

確認你GPUCUDA的計算能力高於3.0(NVIDIA官方支援的對照表這裡)

一、安裝python

windows環境下安裝anaconda。因為tensorflow

只支援python3.5,因此我在電腦上安裝了兩個anacondapython2.7python3.5

python2.7的安裝目錄:

 

Python3.5的安裝目錄:

 

安裝完成後,使用activate啟用python35

輸入activate py3.5可以看到 命令列行前面有個(py3.5),這個就是當前執行環境

接著執行python可以看到python版本顯示為3.5.2

 

如果想返回預設的python 2.7環境,先退出python然後執行deactivate py3.5

執行python,如圖所示,python版本為2.7.12,命令列前面的(py3.5)也不見了,說明已經退出

py3.5環境

 

既然可以在新建的環境下安裝Python 3.5,那我在這個環境下在安裝一個Anaconda3怎麼樣,這樣同時安裝了python 3.5jupyter notebook以及Spyder,嘗試了一下,果然可以!我沒做這步

二、安裝tensorflow

完成上面python3.5安裝後就很簡單了,直接pip安裝就行。

先啟用python3.5的環境:activate py3.5

我原來安裝的是cpu版本的tensorflow,所以先解除安裝pip uninstall tensorflow


注:如果機子顯示卡較低不支援GPU加速就安裝CPU版本,本文安裝的是GPU版本

輸入:

pip install tensorflow-gpu 安裝,如果提示pip版本問題,按提示更新到最新版本即可

由於第一次網路不好出現:

 

換網後:

 

最終安裝成功:

 

三、安裝cuda8.0

雙擊,給定解壓路徑:

 

由於我的機子以前裝了cuda6.5,所以先用360cuda6.5解除安裝了。

解除安裝完成後,重新雙擊setup.exe,然後跳出安裝介面:

 

由於上一版本沒有刪乾淨,但是安裝過程會自己刪除

 

安裝過程:

 

安裝結束

開始驗證

 

編譯檔案:

 


 

四、安裝cuDNN5.1

官網上下載:

 

(1)、解壓縮:會生成cuda/includecuda/libcuda/bin三個目錄;

(2)、分別將cuda/includecuda/libcuda/bin三個目錄中的內容拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5對應的includelibbin目錄下即可。

在系統環境變數CUDA_PATH中新增兩個變數:

CUDA_PATHD:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,但是這樣不能直接訪問到binlib\x64下的程式包:

D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin;

D:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

五、tensorflow-gpu測試

測試一:

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')

sess = tf.Session()