win10 安裝tensorflow-gpu 1.8

其實windows上安裝tensorflow-gpu版相當(dāng)簡單腌且,并不如網(wǎng)上所說的那么麻煩肆汹,只要各個工具的版本正確即可(這里推薦多查閱官方文檔)

環(huán)境準(zhǔn)備

按照tensorflow官網(wǎng)所述品追,CUDA版本需為9.0,下圖展示了Nvidia官網(wǎng)所示的與CUDA 版本兼容的驅(qū)動版本可婶,因此這里需要將驅(qū)動更新到385以上:

CUDA Toolkit and Compatible Driver Version

安裝CUDA 9.0

tensorflow-gpu 1.8 兼容的CUDA版本為9.0稍计,推薦安裝CUDA 9.0本地版,如果你未安裝vs等缀,則在安裝的過程中需取消以下選項

  • [ ] Visual Studio Integration

若你需要vs進行cuda開發(fā)枷莉,則務(wù)必預(yù)先安裝vs,然后在安裝選項中勾選上述選項尺迂。

安裝cudnn 7.1

cudnn 7.0 存在bug笤妙,這里需使用7.1版本冒掌,下載可能需要進行登錄,登錄后下載 cuDNN v7.1.4 (May 16, 2018), for CUDA 9.0蹲盘。下載完成后將文件解壓到CUDA的安裝目錄下

驗證安裝

以上兩步完成后股毫,在命令行中鍵入以下命令查看cuda是否安裝成功,若顯示版本號則安裝成功

nvcc -V

安裝tensorflow-gpu

安裝python

python推薦使用Anaconda進行安裝,python安裝步驟如下:

  • 下載Anaconda

  • 創(chuàng)建python3.5運行環(huán)境

    conda create -n tensorflow pip python=3.5 
    

安裝tensorflow-gpu

  • 激活python3.5環(huán)境

    activate tensorflow
    
  • 輸入以下指令安裝tensorflow-gpu

    pip install -i https://pypi.tuna.tsinghua.edu.cn/simple --ignore-installed --upgrade tensorflow-gpu 
    

驗證安裝

安裝完成后在命令行運行以下指令召衔,以驗證安裝

  • 進入python命令行

    python 
    
  • 執(zhí)行python程序進行驗證

    import tensorflow as tf
    hello = tf.constant('Hello, TensorFlow!')
    sess = tf.Session()
    print(sess.run(hello))
    

若執(zhí)行過程未報錯铃诬,那么恭喜你,你已成功安裝好了tensorflow-gpu 1.8薄嫡。若發(fā)生錯誤請參考官方的常見錯誤說明氧急。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市毫深,隨后出現(xiàn)的幾起案子吩坝,更是在濱河造成了極大的恐慌,老刑警劉巖哑蔫,帶你破解...
    沈念sama閱讀 221,406評論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件钉寝,死亡現(xiàn)場離奇詭異,居然都是意外死亡闸迷,警方通過查閱死者的電腦和手機嵌纲,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,395評論 3 398
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來腥沽,“玉大人逮走,你說我怎么就攤上這事〗裱簦” “怎么了师溅?”我有些...
    開封第一講書人閱讀 167,815評論 0 360
  • 文/不壞的土叔 我叫張陵,是天一觀的道長盾舌。 經(jīng)常有香客問我墓臭,道長,這世上最難降的妖魔是什么妖谴? 我笑而不...
    開封第一講書人閱讀 59,537評論 1 296
  • 正文 為了忘掉前任窿锉,我火速辦了婚禮,結(jié)果婚禮上膝舅,老公的妹妹穿的比我還像新娘嗡载。我一直安慰自己,他們只是感情好仍稀,可當(dāng)我...
    茶點故事閱讀 68,536評論 6 397
  • 文/花漫 我一把揭開白布鼻疮。 她就那樣靜靜地躺著,像睡著了一般琳轿。 火紅的嫁衣襯著肌膚如雪判沟。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,184評論 1 308
  • 那天崭篡,我揣著相機與錄音挪哄,去河邊找鬼。 笑死琉闪,一個胖子當(dāng)著我的面吹牛迹炼,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播颠毙,決...
    沈念sama閱讀 40,776評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼斯入,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了蛀蜜?” 一聲冷哼從身側(cè)響起刻两,我...
    開封第一講書人閱讀 39,668評論 0 276
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎滴某,沒想到半個月后磅摹,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 46,212評論 1 319
  • 正文 獨居荒郊野嶺守林人離奇死亡霎奢,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,299評論 3 340
  • 正文 我和宋清朗相戀三年户誓,在試婚紗的時候發(fā)現(xiàn)自己被綠了。 大學(xué)時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片幕侠。...
    茶點故事閱讀 40,438評論 1 352
  • 序言:一個原本活蹦亂跳的男人離奇死亡帝美,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出晤硕,到底是詐尸還是另有隱情悼潭,我是刑警寧澤,帶...
    沈念sama閱讀 36,128評論 5 349
  • 正文 年R本政府宣布窗骑,位于F島的核電站女责,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏创译。R本人自食惡果不足惜抵知,卻給世界環(huán)境...
    茶點故事閱讀 41,807評論 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望软族。 院中可真熱鬧刷喜,春花似錦、人聲如沸立砸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,279評論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽颗祝。三九已至浊闪,卻和暖如春恼布,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背搁宾。 一陣腳步聲響...
    開封第一講書人閱讀 33,395評論 1 272
  • 我被黑心中介騙來泰國打工折汞, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人盖腿。 一個月前我還...
    沈念sama閱讀 48,827評論 3 376
  • 正文 我出身青樓爽待,卻偏偏與公主長得像,于是被迫代替她去往敵國和親翩腐。 傳聞我的和親對象是個殘疾皇子鸟款,可洞房花燭夜當(dāng)晚...
    茶點故事閱讀 45,446評論 2 359

推薦閱讀更多精彩內(nèi)容