人類史上最簡單的Ubuntu 16.04 安裝tensorflow-gpu教程

最近看了很多網(wǎng)上的ubuntu安裝tensorflow-gpu版本教程偿曙,感覺很復雜州既。

主要問題其實還是cuda cudnn 和tensorflow-gpu版本之間兼容的問題谜洽,不同的tf對應不同的cudnn和cuda。下面介紹本人親測成功的方案

ubuntu16.04 (系統(tǒng)版本無所謂吴叶,都行)

anaconda 3

gpu? gtx1063

cuda? 9.0??

cudnn 7.0? ? (for cuda 9.0)

安裝顯卡驅動

右上角-->系統(tǒng)設置-->軟件和更新-->附加驅動


然后顯卡驅動勾上阐虚,點應用更改

等下載完后重啟一遍,顯卡驅動就安裝好了蚌卤,下載terminal里面輸入nvidia-smi 來查看


這樣就算安裝好了

安裝cuda

本次教程用的版本是cuda-9.0

傳送門:cuda下載地址


這樣選


我只下了這個实束,有需要的可以全都是下

下載完了之后奥秆,terminal cd到下載目錄下面輸入一下命令進行安裝

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda

然后就完成了

這里可以設置環(huán)境變量也可以不設置環(huán)境變量(不用設置也能使用tensorflow-gpu)

還是順便說一下:

export CUDA_HOME=/usr/local/cudaexport PATH=$PATH:$CUDA_HOME/binexport LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

安裝cudnn

這個東西下載之前要注冊登錄一下

傳送門:cudnn下載傳送門

下載如圖:


下這個

現(xiàn)在完之后同理,手動解壓出來吧磕洪,不用命令行了吭练,然后cd到解壓目錄下面,輸入一下命令行,將文件復制到cuda的安裝目錄下(這些代碼是復制粘貼的析显,親測有效鲫咽,一行一行執(zhí)行,不要著急):

sudo cp cuda/include/cudnn.h /usr/local/cuda/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*


環(huán)境變量配置

在terminal中輸入

sudo gedit ~/.bash_profile?

在文本中輸入一下環(huán)境變量

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64"

export CUDA_HOME=/usr/local/cuda-9.0


這樣谷异,兩行分尸,注意cuda是什么版本,環(huán)境變量中就設置什么歹嘹,例如我的是cuda-9.0

最后保存生效在terminal中輸入 source ~/.bash_profile


安裝tensorflow-gpu

本人安裝了anaconda集成環(huán)境箩绍,直接用pip進行安裝,目前tensorflow-gpu最新版本是1.7尺上,但是安裝好了無法使用材蛛,本人降到1.5版本,可以正常使用

在terminal中輸入?

pip install tensorflow-gpu==1.5

速度太慢的同學自行百度:pip 阿里源 清華源等等加速下載

安裝好了然后利用以下代碼進行測試:

import tensorflowas tf

matrix1 = tf.constant([[3., 3.]])

matrix2 = tf.constant([[2.],[2.]])

product = tf.matmul(matrix1, matrix2)

sess = tf.Session()

result = sess.run(product)

print (result)

效果如圖:


顯示了gpu版本顯存等信息怎抛,并出了結果


總結


親測可用的版本卑吭,結合了網(wǎng)上很多大神的方法,自己摸索出來的马绝,不需要什么編譯安裝豆赏,不需要什么降版本gcc,省去了很多麻煩富稻,所有安裝包之類的下載之后掷邦,安裝過程不超過20分鐘

本人第一次寫簡書,如果有寫的不好的地方或者更方便的安裝方法椭赋,可以一起討論

?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末抚岗,一起剝皮案震驚了整個濱河市,隨后出現(xiàn)的幾起案子哪怔,更是在濱河造成了極大的恐慌苟跪,老刑警劉巖,帶你破解...
    沈念sama閱讀 217,509評論 6 504
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件蔓涧,死亡現(xiàn)場離奇詭異件已,居然都是意外死亡,警方通過查閱死者的電腦和手機元暴,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 92,806評論 3 394
  • 文/潘曉璐 我一進店門篷扩,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人茉盏,你說我怎么就攤上這事鉴未∈嘣” “怎么了?”我有些...
    開封第一講書人閱讀 163,875評論 0 354
  • 文/不壞的土叔 我叫張陵铜秆,是天一觀的道長淹真。 經(jīng)常有香客問我,道長连茧,這世上最難降的妖魔是什么核蘸? 我笑而不...
    開封第一講書人閱讀 58,441評論 1 293
  • 正文 為了忘掉前任,我火速辦了婚禮啸驯,結果婚禮上闷煤,老公的妹妹穿的比我還像新娘帚戳。我一直安慰自己,他們只是感情好址愿,可當我...
    茶點故事閱讀 67,488評論 6 392
  • 文/花漫 我一把揭開白布缀遍。 她就那樣靜靜地躺著落塑,像睡著了一般乒省。 火紅的嫁衣襯著肌膚如雪相恃。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 51,365評論 1 302
  • 那天距淫,我揣著相機與錄音杆故,去河邊找鬼。 笑死溉愁,一個胖子當著我的面吹牛,可吹牛的內(nèi)容都是我干的饲趋。 我是一名探鬼主播拐揭,決...
    沈念sama閱讀 40,190評論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼奕塑!你這毒婦竟也來了堂污?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 39,062評論 0 276
  • 序言:老撾萬榮一對情侶失蹤龄砰,失蹤者是張志新(化名)和其女友劉穎盟猖,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體换棚,經(jīng)...
    沈念sama閱讀 45,500評論 1 314
  • 正文 獨居荒郊野嶺守林人離奇死亡式镐,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 37,706評論 3 335
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了固蚤。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片娘汞。...
    茶點故事閱讀 39,834評論 1 347
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖夕玩,靈堂內(nèi)的尸體忽然破棺而出你弦,到底是詐尸還是另有隱情惊豺,我是刑警寧澤,帶...
    沈念sama閱讀 35,559評論 5 345
  • 正文 年R本政府宣布禽作,位于F島的核電站尸昧,受9級特大地震影響,放射性物質發(fā)生泄漏旷偿。R本人自食惡果不足惜烹俗,卻給世界環(huán)境...
    茶點故事閱讀 41,167評論 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望狸捅。 院中可真熱鬧衷蜓,春花似錦、人聲如沸尘喝。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,779評論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽朽褪。三九已至置吓,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間缔赠,已是汗流浹背衍锚。 一陣腳步聲響...
    開封第一講書人閱讀 32,912評論 1 269
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留嗤堰,地道東北人戴质。 一個月前我還...
    沈念sama閱讀 47,958評論 2 370
  • 正文 我出身青樓,卻偏偏與公主長得像踢匣,于是被迫代替她去往敵國和親告匠。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 44,779評論 2 354

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