安裝tensorflow gpu版本

Step1 安裝CUDA8.0

進入這個雲盤地址下載，密碼5aoc 進行CUDA8.0下載。下載完成後解壓，打開exe文件直接按照默認進行安裝，安裝步驟比較繁瑣，時間消耗較長，需耐心等待。

Step2 安裝cuDNN6.0

進入英偉達官網 找到對應版本的cuDNN進行下載，進行簡單的註冊後，可以進入下載頁面

選擇Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0進行下載。下載完畢後，得到一個壓縮包，解壓後將壓縮包中bin、lib、include三個文件夾復制，粘貼到CUDA8.0的安裝目錄下。
如果是按默認方式安裝的CUDA8.0，那麽安裝目錄應該為C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0 。
之後將C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\extras\CUPTI\libx64 與C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64添加到環境變量PATH中。

Step3 安裝Anaconda3.6

進入Anaconda 下載3.6版本，其中自帶python3.6。

Step4 安裝tensorflow

進入Step3安裝完成的Anaconda Prompt, 輸入python 進入python交互環境後，輸入pip3 install --upgrade tensorflow-gpu 回車，即進行tensorflow gpu版本的下載。
等待下載完成，用樣例測試

import tensorflow as tf
a = tf.random_normal((100, 100))
b = tf.random_normal((100, 500))
c = tf.matmul(a, b)
sess = tf.InteractiveSession()
sess.run(c)
 

查看輸出結果

即為成功。

tensorflow win10 系統下安裝