GPU+Tensorflow+Keras(tensorflow/theano)

服務(wù)器可用環(huán)境:

anaconda 安裝的python2.7+cuda8.0+tensorflow1.4+keras2.0.8

centos7上安裝cuda8.0:

https://blog.csdn.net/zhangtong11111/article/details/79143279

安裝tensorflow-gpu和keras:

conda install keras-gpu

小心自己搭配的tensorflow版本和keras版本不一致,有的函數(shù)會(huì)報(bào)錯(cuò)虽画。

檢測tensorflow是否在使用gpu:

import tensorflow as tf?

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

或者:

跑計(jì)算量大的代碼冤馏,通過 nvidia-smi 命令查看gpu的內(nèi)存使用量(不是一開始執(zhí)行程序就能看到的反症,要開始train網(wǎng)絡(luò)了才能看到)

成功使用GPU

開始train時(shí),會(huì)打印包含cuda/gpu的日志:

代碼運(yùn)行日志

測試python程序:

https://blog.csdn.net/freewebsys/article/details/81277857


Tip:

整個(gè)環(huán)境的搭建调煎,用了接近一天的時(shí)間呢,其中遇到了很多問題:

1.cuda版本

我們要選用和tensorflow相匹配的版本,剛開始我一直用的是最新版cuda10祝拯,然而每次在運(yùn)行程序時(shí)提示:沒有libcublas.so.9.0類似的錯(cuò)誤。這是因?yàn)榘姹静恢С帧K赃€是下載8.0吧佳头,大多是穩(wěn)的鹰贵。

其次,官網(wǎng)建議下載rpm的版本安裝康嘉,然而我嚴(yán)格按照官網(wǎng)安裝了碉输,還是不成功,所以還是用run文件安裝吧亭珍。按上述博主的方式安裝敷钾,沒有馬達(dá)!

2.tensorflow+keras

這倆版本也要一致的块蚌。不能差太大闰非,否則會(huì)報(bào)一些函數(shù)錯(cuò)誤。用默認(rèn)的conda install keras-gpu就可以完美安裝兩者峭范,不需要先安裝tensorflow哦财松。

3.theano+keras

keras的默認(rèn)后端是tensorflow,如果要使用theano作為后端纱控,需要安裝theano辆毡,然后進(jìn)行配置。命令如下:

安裝theano:conda install then

配置keras:vim ~/.keras/keras.json

在打開的文件中甜害,將tensorflow改為theano即可舶掖,當(dāng)然也可以支持后端cntk。

配置keras后端

4.嚴(yán)格驗(yàn)證

一定要嚴(yán)格用命令nvcc -v驗(yàn)證cuda的正確安裝尔店。

一定要確認(rèn)tensorflow使用了gpu訓(xùn)練眨攘。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市嚣州,隨后出現(xiàn)的幾起案子鲫售,更是在濱河造成了極大的恐慌,老刑警劉巖该肴,帶你破解...
    沈念sama閱讀 212,454評(píng)論 6 493
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件情竹,死亡現(xiàn)場離奇詭異,居然都是意外死亡匀哄,警方通過查閱死者的電腦和手機(jī)秦效,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,553評(píng)論 3 385
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來涎嚼,“玉大人阱州,你說我怎么就攤上這事》ㄌ荩” “怎么了苔货?”我有些...
    開封第一講書人閱讀 157,921評(píng)論 0 348
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我蒲赂,道長,這世上最難降的妖魔是什么刁憋? 我笑而不...
    開封第一講書人閱讀 56,648評(píng)論 1 284
  • 正文 為了忘掉前任滥嘴,我火速辦了婚禮,結(jié)果婚禮上至耻,老公的妹妹穿的比我還像新娘若皱。我一直安慰自己,他們只是感情好尘颓,可當(dāng)我...
    茶點(diǎn)故事閱讀 65,770評(píng)論 6 386
  • 文/花漫 我一把揭開白布走触。 她就那樣靜靜地躺著,像睡著了一般疤苹。 火紅的嫁衣襯著肌膚如雪互广。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 49,950評(píng)論 1 291
  • 那天卧土,我揣著相機(jī)與錄音惫皱,去河邊找鬼。 笑死尤莺,一個(gè)胖子當(dāng)著我的面吹牛旅敷,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播颤霎,決...
    沈念sama閱讀 39,090評(píng)論 3 410
  • 文/蒼蘭香墨 我猛地睜開眼媳谁,長吁一口氣:“原來是場噩夢(mèng)啊……” “哼!你這毒婦竟也來了友酱?” 一聲冷哼從身側(cè)響起晴音,我...
    開封第一講書人閱讀 37,817評(píng)論 0 268
  • 序言:老撾萬榮一對(duì)情侶失蹤,失蹤者是張志新(化名)和其女友劉穎粹污,沒想到半個(gè)月后段多,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 44,275評(píng)論 1 303
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡壮吩,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 36,592評(píng)論 2 327
  • 正文 我和宋清朗相戀三年进苍,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片鸭叙。...
    茶點(diǎn)故事閱讀 38,724評(píng)論 1 341
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡觉啊,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出沈贝,到底是詐尸還是另有隱情杠人,我是刑警寧澤,帶...
    沈念sama閱讀 34,409評(píng)論 4 333
  • 正文 年R本政府宣布,位于F島的核電站嗡善,受9級(jí)特大地震影響辑莫,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜罩引,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 40,052評(píng)論 3 316
  • 文/蒙蒙 一各吨、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧袁铐,春花似錦揭蜒、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,815評(píng)論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至洒缀,卻和暖如春瑰谜,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背帝洪。 一陣腳步聲響...
    開封第一講書人閱讀 32,043評(píng)論 1 266
  • 我被黑心中介騙來泰國打工似舵, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人葱峡。 一個(gè)月前我還...
    沈念sama閱讀 46,503評(píng)論 2 361
  • 正文 我出身青樓砚哗,卻偏偏與公主長得像,于是被迫代替她去往敵國和親砰奕。 傳聞我的和親對(duì)象是個(gè)殘疾皇子蛛芥,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 43,627評(píng)論 2 350

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

  • 2016年7月,縣里舉辦了電商培訓(xùn)班军援,我在開始了解電商仅淑,以前總以為電商不就像商店一樣賣東西嗎?不胸哥,通過學(xué)習(xí)涯竟,我才知...
    小鎮(zhèn)名家閱讀 371評(píng)論 0 0
  • 仔仔終于返校了,向老師交待和保證以后不會(huì)再犯錯(cuò)誤了空厌。沒想到下午姥爺紙去喂鴿子摔到了庐船,老媽打了120,我陪同...
    冉303656閱讀 166評(píng)論 0 0
  • 這是我今天早上收藏的一個(gè)朋友圈嘲更,我把他復(fù)述一遍筐钟,不再是學(xué)生時(shí)代的我,以前視頻學(xué)到一樣?xùn)|西赋朦,書本上學(xué)到一樣?xùn)|西篓冲,馬上...
    中土閱讀 191評(píng)論 0 1