1. 程式人生 > >Win10系統下安裝cudnn6.0+CUDA8.0+anaconda+tensorflow GPU版本

Win10系統下安裝cudnn6.0+CUDA8.0+anaconda+tensorflow GPU版本

cudnn6安裝包: https://pan.baidu.com/s/1ko5kTFj5hTNrGEBpmqgs-A 密碼:jpw5
之前在win10下一直配置不好TensorFlow(GPU)版本,苦苦掙扎幾天後無奈轉向在ubantu 14.04下安裝,安裝成功後近日又想在Windows上再試一次,結果很順利,幾個小時便安裝好了TensorFlow GPU版本。本文主要寫基於anaconda的TensorFlow安裝,如有錯誤請聯絡作者修改~~~(小白一枚)
環境:win10(64位)
GPU: GTX960
此外還要確保電腦已安裝有visual studio2015或2013版本等。

第一步:安裝Anaconda
Anaconda裡面集成了很多關於python科學計算的第三方庫,主要是安裝方便,在官網(

https://www.continuum.io/downloads)上即可下載。

由於在Windows下TensorFlow目前只支援Python3.5 64-bit版本,為了安裝方便,建議安裝anaconda4.2.0版本,若為其他版本,需conda建立一個Python3.5的環境。
目前anaconda已更新至4.4.0版本,若需安裝anaconda 4.2.0版本,在圖示頁面點選:Use thesezippedWindows installers,彈出了anaconda的歷史版本供下載,選擇相應版本即可。

第二步:安裝CUDA

CUDA分network版本與local版本,建議下載local版本,因為network版本是在安裝過程中下載的,選擇本地版較好,不過在官網下載容易出錯(我下了幾次均是下載到一半提示任務出錯),建議在百度雲下載。
下載之後安裝即可。

第三步:下載cudnn
初次下載cudnn需要在網上填寫一堆問卷,耐心填吧。(下載地址:https://developer.nvidia.com/cudnn)下載好後無須安裝。
將包內三個檔案複製到cuda相應位置,如圖示。

同時還需將bin目錄、lib\64目錄新增到環境變數path中,否則後面呼叫時會報錯。

第四步:安裝Tensorflow(GPU)
開啟Anaconda Prompt,輸入:pip install tensorflow-gpu安裝即可。

如果提示pip版本需更新,按提示操作即可。

經過上面幾步,TensorFlow的GPU版本就安裝好了,開啟Anaconda Prompt,輸入python或spyder或jupyter notebook即可開車了~~~。
下面驗證下是否成功:
import tensorflow as tf

hello =tf.constant(“Hello!TensorFlow”)
sess = tf.Session()
print(sess.run(hello))

Creates a graph.

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0,6.0], shape=[2, 3], name=’a’)
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0,6.0], shape=[3, 2], name=’b’)
c = tf.matmul(a, b)

Creates a session withlog_device_placement set to True.

sess =tf.Session(config=tf.ConfigProto(log_device_placement=True))

Runs the op.

print(sess.run(c))