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

GTX 1080Ti

配置深度學(xué)習(xí)主機(jī)與環(huán)境(TensorFlow+1080Ti):
(一)硬件選購與主機(jī)組裝
(二)Win10&Ubuntu雙系統(tǒng)與顯卡驅(qū)動安裝
(三)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ā)展變化,沒有必要每次都為不同的硬件結(jié)構(gòu)重新編碼,而CUDA就是提供了一種可擴(kuò)展的編程模型,使得已經(jīng)寫好的CUDA代碼可以在任意數(shù)量核心的GPU上運(yùn)行。
CUDA最主要的包含兩個(gè)方面:一個(gè)是ISA指令集架構(gòu);第二硬件計(jì)算引擎;實(shí)際上是硬件和指令集。 也就是說我們可以把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ū)動

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

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

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

2. 安裝CUDA

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


    安裝指南
  • 打開終端,輸入:sudo sh cuda_8.0.61_375.26_linux.run,在下載選項(xiàng)中,也有說明如何安裝;
    `sudo sh cuda_8.0.61_375.26_linux.run`
  • 如果出現(xiàn)大段廢話,可以使用 Ctrl+C 跳過;
  • 進(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ū)動,無需重新安裝。
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 測試
    運(yùn)行 CUDA Samples 檢測 CUDA 是否順利安裝:
  • 打開終端,切換到位置:~/ NVIDIA_CUDA-8.0_Samples
  • 輸入:make
    CUDA Samples 測試
  • 編譯時(shí)間較長。若沒有報(bào)錯(cuò),則將新編譯為二進(jìn)制文件,默認(rèn)存放在 ~/NVIDIA_CUDA-7.5_Samples/bin 中。
  • 打開終端,切換到位置:~/NVIDIA_CUDA-8.0_Samples/bin/x86_64/linux/release
  • 終端輸入:./deviceQuery,如果顯示類似下圖,則說明CUDA安裝且配置成功!
    CUDA安裝且配置成功!

3. 安裝cuDNN

  1. 下載cuDNN,下載之前需要注冊Nvidia開發(fā)者帳號,按要求注冊即可。
  • 莫名其妙在我的筆記本上無法登錄...在主機(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ò)誤(參見本文最后一張圖,是安裝cuDNN v6時(shí)的截圖,其中包含libcudnn.so.6文件)。

  • stcokoverflow 中也出現(xiàn)了相關(guā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ù)制粘貼,手動復(fù)制粘貼也是沒問題的。到此cuDNN安裝完畢。


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

4. 參考資料

  1. 知乎:Ubuntu 16.04.2 + GTX1080 + Python3 TensorFlow 配置
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。
  • 序言:七十年代末,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,119評論 6 531
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件,死亡現(xiàn)場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī),發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 98,382評論 3 415
  • 文/潘曉璐 我一進(jìn)店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,038評論 0 373
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經(jīng)常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 62,853評論 1 309
  • 正文 為了忘掉前任,我火速辦了婚禮,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 71,616評論 6 408
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 55,112評論 1 323
  • 那天,我揣著相機(jī)與錄音,去河邊找鬼。 笑死,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,192評論 3 441
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 42,355評論 0 288
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體,經(jīng)...
    沈念sama閱讀 48,869評論 1 334
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 40,727評論 3 354
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 42,928評論 1 369
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,467評論 5 358
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 44,165評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,570評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,813評論 1 282
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 51,585評論 3 390
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個(gè)殘疾皇子,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 47,892評論 2 372

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