Windows安裝TensorFlow-GPU教程

本文介紹 tensorflow-gpu 1.4.0版本在Windows下的安裝教程,并涉及一些必備軟件的安裝。

準備步驟

簡單的說,我們需要 Windows系統 + Anaconda(Python 3.6) + Visual Studio 2015 + [顯卡驅動] + CUDA 8.0 + CuDnn 6.0。具體安裝流程在第二部分介紹。

系統要求

Windows版本:Windows 7,Windows 8,Windows 10,Windows Server 12/16

Anaconda

使用Anaconda主要是方便搭建和切換不同的Python開發環境,具體安裝步驟這里不給出,沒有安裝的可以參考Anaconda介紹、安裝及使用教程

vs 2015

Visual Studio 2015主要是提供C++的開發編譯環境,不然cudacuDnn庫將無法正常解析。
下載免費的Community版本即可,下載鏈接 提取碼:q5xi 。

顯卡要求

具有計算能力3.5或更高版本的NVIDIA?GPU卡。請注意,AMD的顯卡不可以使用英偉達開發的CUDA
顯卡計算能力的查詢可移步到https://developer.nvidia.com/cuda-gpus自行查詢。比如筆者的筆記本顯卡是GeForce GTX 950M

GeForce GTX 950M的計算能力為5.0

驅動程序

另外,如果電腦沒有安裝顯卡的驅動程序,可以移步NVIDIA? GPU 驅動程序自行下載對應型號的驅動程序。

驅動程序下載

CUDA版本

不同版本的TensorFlow需要指定版本的CUDA的支持,比如本文的tensorflow-gpu 1.4.0版本,需要CUDA 8.0的配合。更多版本對應關系詳見文末的附錄。

CUDA下載鏈接:CUDA? Toolkit

CUDA 8.0

CUDA 8.0下載頁面

CuDnn版本

CuDnn版本: **CuDnn 6.0 **for CUDA8.0。下載時請對應的你的操作系統選擇合適的版本。

請特別注意下載CuDnn前必須注冊英偉達社區的會員,https://developer.nvidia.com/rdp/form/cudnn-download-survey,雖然說是英文的但非常簡單。

查看其它版本

下載CuDnn 6.0

所有軟件

所有軟件

安裝步驟

請盡量按照本文的安裝順序進行安裝,否則有可能運行失敗!最重要的是,安裝CUDA必須在安裝Visual Studio之后,這兩個順序不可以調換。

安裝驅動程序

如果已經安裝,請跳到下一步。
驅動程序安裝比較容易,雙擊安裝程序,一直點下一步即可。

安裝Visual Studio 2015

安裝VS2015沒什么難度,只有一個選項需要自定義,別的選項都可以使用默認值。VS2015的默認安裝不包括C++的編譯器,必須手動勾選Visual C++,不然會面臨后續的CUDA編譯錯誤。


安裝VS2015 勾選Visual C++

安裝CUDA

安裝CUDA前請務必確認VS2015安裝成功!這個時候你需要安裝CUDA,雙擊我們下載的安裝文件即可,雙擊后會出現下圖,要求選擇一個目錄用于存放NVIDIA CUDA Toolkit安裝器,隨便選個目錄即可,安裝完會自動刪除。

image.png

安裝成功后,會出現下圖:


image.png

驗證CUDA安裝成功

打開命令行,也就是cmd然后輸入“nvcc -V”,如果安裝正確的話你應該看到下圖的輸出,輸出中顯示了CUDA的版本是release 8.0。


CUDA 8.0安裝成功

確認系統環境變量

CUDA安裝成功后,會自動添加安裝目錄到系統環境變量,所以需要檢查C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin是否已加到系統環境變量,若沒有則需要手動添加。

image.png

安裝CuDnn

解壓縮我們下載的CuDnn文件,得到3個文件夾:bin, include, lib。

image.png

image.png

將這個三個文件夾復制到:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,如 圖:
image.png

創建Python環境

Python環境的創建,使用的是conda命令。打開Anaconda命令行窗口,如圖:

Anaconda Prompt

Anaconda命令行窗口

# 創建Python環境,名稱為tensorflow,Python版本為3.6
$ conda create --name tensorflow python=3.6

# 安裝好后,使用activate激活環境
$ activate tensorflow

# 若需要退出環境,使用deactivate命令
$ conda deactivate

安裝tensorflow-gpu

# 使用pip安裝tensorflow-gpu,版本為1.4.0,并國內的豆瓣pip源
$ pip install tensorflow-gpu==1.4 -i https://pypi.douban.com/simple/

# 安裝Keras(可選)
$ pip install keras==2.1.3 -i https://pypi.douban.com/simple/

驗證TensorFlow GPU

# 激活tensorflow環境
$ activate tensorflow
# 進入Python的 interactive shell
$ python
# 第一個TensorFlow程序
>>> import tensorflow as tf
>>> 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)
>>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
>>> print(sess.run(c))

如果出現類似下圖圈中的輸出,則代表 TensorFlow GPU是安裝成功的。


驗證TensorFlow GPU

附錄

附錄A:tensorflow CUDA cudnn 版本對應關系

Windows下tensorflow CUDA cudnn 版本對應關系

linux下tensorflow CUDA cudnn 版本對應關系

附錄B:TensorFlow示例運行

TensorFlow GitHub官方倉庫

下載相應的代碼,可以直接運行samples里面實例。

附錄C:cuda 降版本

windows 7 下cuda 9.0 卸載、cuda8.0 安裝

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,546評論 6 533
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,570評論 3 418
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,505評論 0 376
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,017評論 1 313
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,786評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,219評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,287評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,438評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 48,971評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,796評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,995評論 1 369
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,540評論 5 359
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,230評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,662評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,918評論 1 286
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,697評論 3 392
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 47,991評論 2 374

推薦閱讀更多精彩內容