配置深度學(xué)習(xí)主機(jī)與環(huán)境(TensorFlow+1080Ti):(三)CUDA與CUDNN安裝

GTX 1080Ti

配置深度學(xué)習(xí)主機(jī)與環(huán)境(TensorFlow+1080Ti):
(一)硬件選購(gòu)與主機(jī)組裝
(二)Win10&Ubuntu雙系統(tǒng)與顯卡驅(qū)動(dòng)安裝
(三)CUDA與CUDNN安裝
(四)基于Anaconda的TensorFlow安裝


0. 概念介紹

CUDA(Compute Unified Device Architecture 統(tǒng)一計(jì)算設(shè)備架構(gòu))

CUDA(Compute Unified Device Architecture),是英偉達(dá)公司推出的一種基于新的并行編程模型和指令集架構(gòu)的通用計(jì)算架構(gòu),它能利用英偉達(dá)GPU的并行計(jì)算引擎僧鲁,比CPU更高效的解決許多復(fù)雜計(jì)算任務(wù)厕诡。
使用CUDA的好處就是透明温艇。根據(jù)摩爾定律GPU的晶體管數(shù)量不斷增多肠虽,硬件結(jié)構(gòu)必然是不斷的在發(fā)展變化醉者,沒(méi)有必要每次都為不同的硬件結(jié)構(gòu)重新編碼刽辙,而CUDA就是提供了一種可擴(kuò)展的編程模型窥岩,使得已經(jīng)寫好的CUDA代碼可以在任意數(shù)量核心的GPU上運(yùn)行。
CUDA最主要的包含兩個(gè)方面:一個(gè)是ISA指令集架構(gòu)宰缤;第二硬件計(jì)算引擎颂翼;實(shí)際上是硬件和指令集。 也就是說(shuō)我們可以把CUDA看做是與X86或者cell類似的架構(gòu)慨灭,但是是基于是GPU朦乏,而不是傳統(tǒng)的CPU。

cuDNN(NVIDIA CUDA? Deep Neural Network library)

The NVIDIA CUDA? Deep Neural Network library (cuDNN) is a GPU-accelerated library of primitives for deep neural networks. cuDNN provides highly tuned implementations for standard routines such as forward and backward convolution, pooling, normalization, and activation layers. cuDNN is part of the NVIDIA Deep Learning SDK.
Deep learning researchers and framework developers worldwide rely on cuDNN for high-performance GPU acceleration. It allows them to focus on training neural networks and developing software applications rather than spending time on low-level GPU performance tuning. cuDNN accelerates widely used deep learning frameworks, including Caffe, Caffe2, TensorFlow, Theano, Torch, and Microsoft Cognitive Toolkit.

cuDNN Accelerated Frameworks

1. 安裝顯卡驅(qū)動(dòng)

在安裝CUDA和cuDNN之前氧骤,需要確保顯卡驅(qū)動(dòng)已經(jīng)安裝好呻疹,其安裝情況可以在 設(shè)置-軟件更新-附加驅(qū)動(dòng) 中查看:

設(shè)置-軟件更新-附加驅(qū)動(dòng)

更多關(guān)于顯卡(文中使用1080Ti)安裝的步驟,可以參考系列文章的第二篇:
配置深度學(xué)習(xí)主機(jī)與環(huán)境(TensorFlow+1080Ti):(二)Win10&Ubuntu雙系統(tǒng)與顯卡驅(qū)動(dòng)安裝

2. 安裝CUDA

  1. 查閱 NVIDIA CUDA Installation Guide for Linux
    英偉達(dá)CUDA安裝指南(Linux)
    對(duì)于后續(xù)的學(xué)習(xí)與工作大有裨益筹陵,包含了CUDA的詳盡安裝說(shuō)明刽锤。當(dāng)然如果你不想閱讀英文镊尺,只想盡快安裝CUDA,可以參照后續(xù)步驟并思。
  • 下載 CUDA Toolkit 安裝包:
    安裝包下載選項(xiàng)
  • 推薦使用第一個(gè)runfile (local) 安裝選項(xiàng)庐氮;
  • 網(wǎng)上不少教程中寫有需要降低gcc版本,實(shí)測(cè)宋彼,無(wú)需降級(jí)弄砍,因此可以忽略;
  • 下載第一個(gè) Base Installer 即可宙暇。
  • 安裝CUDA8.0


    安裝指南
  • 打開終端输枯,輸入:sudo sh cuda_8.0.61_375.26_linux.run,在下載選項(xiàng)中占贫,也有說(shuō)明如何安裝桃熄;
    `sudo sh cuda_8.0.61_375.26_linux.run`
  • 如果出現(xiàn)大段廢話,可以使用 Ctrl+C 跳過(guò)型奥;
  • 進(jìn)入具體的安裝選項(xiàng):


    安裝選項(xiàng)
  • 注意第二個(gè)選項(xiàng) Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26? 須選擇否瞳收,因?yàn)樵谥耙呀?jīng)安裝好了顯卡驅(qū)動(dòng),無(wú)需重新安裝厢汹。
Do you accept the previously read EULA?
accept/decline/quit: accept
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
(y)es/(n)o/(q)uit: n
Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y
Enter Toolkit Location
 [ default is /usr/local/cuda-8.0 ]:  
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
Install the CUDA 8.0 Samples?
(y)es/(n)o/(q)uit: y 
Enter CUDA Samples Location
 [ default is /home/dexter ]:
  • 添加環(huán)境變量:
    • 在終端中輸入:
gedit ~/.bashrc
  • 在打開的文件中寫入:
export PATH="$PATH:/usr/local/cuda-8.0/bin"
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64"
  • 保存并關(guān)閉文件螟深, 在終端中輸入:
source ~/.bashrc

使環(huán)境更改生效。

  • 在終端中輸入nvidia-smi烫葬,可以查詢顯卡狀態(tài)界弧。
    查詢顯卡狀態(tài)
  • CUDA Samples 測(cè)試
    運(yùn)行 CUDA Samples 檢測(cè) CUDA 是否順利安裝:
  • 打開終端,切換到位置:~/ NVIDIA_CUDA-8.0_Samples
  • 輸入:make
    CUDA Samples 測(cè)試
  • 編譯時(shí)間較長(zhǎng)搭综。若沒(méi)有報(bào)錯(cuò)垢箕,則將新編譯為二進(jìn)制文件,默認(rèn)存放在 ~/NVIDIA_CUDA-7.5_Samples/bin 中兑巾。
  • 打開終端条获,切換到位置:~/NVIDIA_CUDA-8.0_Samples/bin/x86_64/linux/release
  • 終端輸入:./deviceQuery,如果顯示類似下圖蒋歌,則說(shuō)明CUDA安裝且配置成功帅掘!
    CUDA安裝且配置成功!

3. 安裝cuDNN

  1. 下載cuDNN堂油,下載之前需要注冊(cè)Nvidia開發(fā)者帳號(hào)修档,按要求注冊(cè)即可。
  • 莫名其妙在我的筆記本上無(wú)法登錄...在主機(jī)的Ubuntu中可以順利登錄...此條純屬吐槽府框,封IP真是夠了...


    IP blocked
  • 登錄后顯示有多個(gè)版本可供選擇萍悴,建議使用 cuDNN v5.1 for CUDA 8.0,因?yàn)樽髡咦铋_始嘗鮮選擇 cuDNN v6.0 for CUDA 8.0 最后安裝失敗寓免,應(yīng)該是支持還不到位癣诱。然后又刪掉cuDNN,再重新安裝(中間刪掉重新安裝cuDNN已經(jīng)安裝好的Anaconda不影響袜香,但是最好重新安裝TensorFlow)撕予。
  • 具體報(bào)錯(cuò)為:
ImportError: libcudnn.so.5: cannot open shared object file: No such file or director
Error

在安裝cuDNN中,可以看到安裝文件的版本為libcudnn.so.6蜈首,所以很明顯是v6版本不支持導(dǎo)致的錯(cuò)誤(參見(jiàn)本文最后一張圖实抡,是安裝cuDNN v6時(shí)的截圖,其中包含libcudnn.so.6文件)欢策。

  • stcokoverflow 中也出現(xiàn)了相關(guān)問(wèn)題:
    ImportError: libcudnn when running a TensorFlow program

    ImportError: libcudnn when running a TensorFlow program

  • 選擇 cuDNN v5.1 for CUDA 8.0 中的 cuDNN v5.1 Library for Linux 下載:

    cuDNN v5.1 Library for Linux

  • 在包含安裝文件的文件夾吆寨,右鍵,打開終端踩寇,輸入如下命令行:

tar xvzf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64
sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*

相當(dāng)于解壓縮+復(fù)制粘貼啄清,手動(dòng)復(fù)制粘貼也是沒(méi)問(wèn)題的。到此cuDNN安裝完畢俺孙。


cuDNN安裝完畢(v6僅作示范辣卒,后面刪除換為了v5)

4. 參考資料

  1. 知乎:Ubuntu 16.04.2 + GTX1080 + Python3 TensorFlow 配置
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市睛榄,隨后出現(xiàn)的幾起案子荣茫,更是在濱河造成了極大的恐慌,老刑警劉巖场靴,帶你破解...
    沈念sama閱讀 206,013評(píng)論 6 481
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件啡莉,死亡現(xiàn)場(chǎng)離奇詭異,居然都是意外死亡旨剥,警方通過(guò)查閱死者的電腦和手機(jī)咧欣,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 88,205評(píng)論 2 382
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來(lái)泞边,“玉大人该押,你說(shuō)我怎么就攤上這事≌笱瑁” “怎么了蚕礼?”我有些...
    開封第一講書人閱讀 152,370評(píng)論 0 342
  • 文/不壞的土叔 我叫張陵,是天一觀的道長(zhǎng)梢什。 經(jīng)常有香客問(wèn)我奠蹬,道長(zhǎng),這世上最難降的妖魔是什么嗡午? 我笑而不...
    開封第一講書人閱讀 55,168評(píng)論 1 278
  • 正文 為了忘掉前任囤躁,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘狸演。我一直安慰自己言蛇,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 64,153評(píng)論 5 371
  • 文/花漫 我一把揭開白布宵距。 她就那樣靜靜地躺著腊尚,像睡著了一般。 火紅的嫁衣襯著肌膚如雪满哪。 梳的紋絲不亂的頭發(fā)上婿斥,一...
    開封第一講書人閱讀 48,954評(píng)論 1 283
  • 那天,我揣著相機(jī)與錄音哨鸭,去河邊找鬼民宿。 笑死,一個(gè)胖子當(dāng)著我的面吹牛像鸡,可吹牛的內(nèi)容都是我干的活鹰。 我是一名探鬼主播,決...
    沈念sama閱讀 38,271評(píng)論 3 399
  • 文/蒼蘭香墨 我猛地睜開眼坟桅,長(zhǎng)吁一口氣:“原來(lái)是場(chǎng)噩夢(mèng)啊……” “哼华望!你這毒婦竟也來(lái)了?” 一聲冷哼從身側(cè)響起仅乓,我...
    開封第一講書人閱讀 36,916評(píng)論 0 259
  • 序言:老撾萬(wàn)榮一對(duì)情侶失蹤赖舟,失蹤者是張志新(化名)和其女友劉穎,沒(méi)想到半個(gè)月后夸楣,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體宾抓,經(jīng)...
    沈念sama閱讀 43,382評(píng)論 1 300
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長(zhǎng)有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 35,877評(píng)論 2 323
  • 正文 我和宋清朗相戀三年豫喧,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了石洗。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片提茁。...
    茶點(diǎn)故事閱讀 37,989評(píng)論 1 333
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡赶袄,死狀恐怖倦西,靈堂內(nèi)的尸體忽然破棺而出俱箱,到底是詐尸還是另有隱情,我是刑警寧澤困肩,帶...
    沈念sama閱讀 33,624評(píng)論 4 322
  • 正文 年R本政府宣布深碱,位于F島的核電站草冈,受9級(jí)特大地震影響篙程,放射性物質(zhì)發(fā)生泄漏枷畏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 39,209評(píng)論 3 307
  • 文/蒙蒙 一虱饿、第九天 我趴在偏房一處隱蔽的房頂上張望拥诡。 院中可真熱鬧触趴,春花似錦、人聲如沸渴肉。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,199評(píng)論 0 19
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽(yáng)宾娜。三九已至批狐,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間前塔,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 31,418評(píng)論 1 260
  • 我被黑心中介騙來(lái)泰國(guó)打工承冰, 沒(méi)想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留华弓,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 45,401評(píng)論 2 352
  • 正文 我出身青樓困乒,卻偏偏與公主長(zhǎng)得像寂屏,于是被迫代替她去往敵國(guó)和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子娜搂,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 42,700評(píng)論 2 345

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