Windows安裝TensorFlow-GPU流程

環(huán)境:

  • Windows 10 x64
  • GTX 960

整體流程并不復雜请毛,可分為如下五步:

  • 安裝NVidia驅動
  • 安裝CUDA 9.0
  • 安裝cuDNN 7.x for CUDA 9.0
  • 安裝Anaconda
  • pip安裝tensorflow-gpu

下面逐一進行記錄

一、安裝NVidia驅動

這里只需要直接百度NVidia部逮,找到NVidia官網(wǎng)院促,下載好驅動安裝就可以了筏养。其實就是在這里搜索對應的顯卡驅動,進行下載常拓。

下載完成后渐溶,重啟一下電腦即可。

二弄抬、安裝CUDA 9.0

本來自己之前安裝的是CUDA 9.1茎辐,但是所有安裝完成后,imort tensorflow時報錯找不到相應的動態(tài)鏈接庫掂恕,細看文件名發(fā)現(xiàn)拖陆,原來tensorflow使用的是CUDA 9.0,所以不得不重新下載安裝懊亡。后來查看官方文檔發(fā)現(xiàn)慕蔚,官方文檔中給出的組件正是 CUDA Toolkit 9.0,嚴格按照官方來斋配,總不會錯孔飒。

官網(wǎng)下載的時候,默認給出的是最新版9.1的下載地址艰争,注意要選擇Legacy Releases找到9.0版本坏瞄,這里直接給出地址 https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

下載好 Base Installer 和 Patch 1,先安裝 Base Installer甩卓,然后安裝 Patch 1即可鸠匀。

安裝完成后,打開命令提示符逾柿,輸入 echo %CUDA_PATH%缀棍,可以看到CUDA_PATH這個環(huán)境變量已經(jīng)設置好了。TensorFlow官網(wǎng)上要求將這個變量加入到%PATH%中机错,加入的方式為:

搜索并打開run > 輸入control sysdm.cpl > 高級 > 環(huán)境變量 > 選中 path 點擊編輯 > 新建 > 輸入%CUDA_PATH% > 確定保存

這樣 CUDA 9.0就已經(jīng)安裝好了爬范。

三、安裝cuDNN v7

這里在安裝的時候也是遇到坑的弱匪,就是 https://developer.nvidia.com/ 這個網(wǎng)站總是顯示正在維護青瀑,因而也找不到下載的地方,后來在知乎上找到了解決方法:

nvidia的cudnn為什么不能下載了? - 瞎折騰的一生的回答 - 知乎

里面截圖中給出的鏈接 https://developer.nvidia.com/rdp/cudnn-download 可以使用斥难,進去后注冊登錄枝嘶,同意用戶協(xié)議即可下載。需要注意的是 cuDNN 有一個版本號哑诊,而同一個版本的 cuDNN 適配了不同版本的 CUDA群扶,這需要和自己安裝的CUDA版本對應起來。所以镀裤,需要下載的是 cuDNN v7.1.1 for CUDA 9.0竞阐。

安裝的過程非常簡單,只需要將cuda文件夾下的內容復制到%CUDA_PATH%下即可.

cuDNN/cuda %CUDA_PATH%
bin/* bin/*
include/* include/*
lib/* lib/*

四淹禾、安裝Anaconda

推薦清華鏡像源,下載速度有保障茴扁。這里下載的是 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.1.0-Windows-x86_64.exe

五铃岔、安裝tensorflow-gpu

1.配置pip源,使用清華鏡像

https://mirrors.tuna.tsinghua.edu.cn/help/pypi/

2.安裝tensorflow

開始 > Anaconda3 (64-bit) > Anaconda Prompt

輸入

pip install tensorflow-gpu
?著作權歸作者所有,轉載或內容合作請聯(lián)系作者
  • 序言:七十年代末峭火,一起剝皮案震驚了整個濱河市毁习,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌卖丸,老刑警劉巖纺且,帶你破解...
    沈念sama閱讀 217,542評論 6 504
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異稍浆,居然都是意外死亡载碌,警方通過查閱死者的電腦和手機,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,822評論 3 394
  • 文/潘曉璐 我一進店門衅枫,熙熙樓的掌柜王于貴愁眉苦臉地迎上來嫁艇,“玉大人,你說我怎么就攤上這事弦撩〔竭洌” “怎么了?”我有些...
    開封第一講書人閱讀 163,912評論 0 354
  • 文/不壞的土叔 我叫張陵益楼,是天一觀的道長猾漫。 經(jīng)常有香客問我,道長感凤,這世上最難降的妖魔是什么悯周? 我笑而不...
    開封第一講書人閱讀 58,449評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮陪竿,結果婚禮上队橙,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好捐康,可當我...
    茶點故事閱讀 67,500評論 6 392
  • 文/花漫 我一把揭開白布仇矾。 她就那樣靜靜地躺著,像睡著了一般解总。 火紅的嫁衣襯著肌膚如雪贮匕。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,370評論 1 302
  • 那天花枫,我揣著相機與錄音刻盐,去河邊找鬼。 笑死劳翰,一個胖子當著我的面吹牛敦锌,可吹牛的內容都是我干的。 我是一名探鬼主播佳簸,決...
    沈念sama閱讀 40,193評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼乙墙,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了生均?” 一聲冷哼從身側響起听想,我...
    開封第一講書人閱讀 39,074評論 0 276
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎马胧,沒想到半個月后汉买,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 45,505評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡佩脊,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 37,722評論 3 335
  • 正文 我和宋清朗相戀三年蛙粘,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片威彰。...
    茶點故事閱讀 39,841評論 1 348
  • 序言:一個原本活蹦亂跳的男人離奇死亡组题,死狀恐怖,靈堂內的尸體忽然破棺而出抱冷,到底是詐尸還是另有隱情崔列,我是刑警寧澤,帶...
    沈念sama閱讀 35,569評論 5 345
  • 正文 年R本政府宣布旺遮,位于F島的核電站赵讯,受9級特大地震影響,放射性物質發(fā)生泄漏耿眉。R本人自食惡果不足惜边翼,卻給世界環(huán)境...
    茶點故事閱讀 41,168評論 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望鸣剪。 院中可真熱鬧组底,春花似錦丈积、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,783評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至厌均,卻和暖如春唬滑,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背棺弊。 一陣腳步聲響...
    開封第一講書人閱讀 32,918評論 1 269
  • 我被黑心中介騙來泰國打工晶密, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人模她。 一個月前我還...
    沈念sama閱讀 47,962評論 2 370
  • 正文 我出身青樓稻艰,卻偏偏與公主長得像,于是被迫代替她去往敵國和親侈净。 傳聞我的和親對象是個殘疾皇子尊勿,可洞房花燭夜當晚...
    茶點故事閱讀 44,781評論 2 354

推薦閱讀更多精彩內容