Ubuntu16.04安裝NVIDIA顯卡驅(qū)動并配置tensorflow-gpu深度學習框架

系統(tǒng):Ubuntu16.04 LTS
顯卡:GeForce GTX 1060

1 安裝NVIDIA顯卡驅(qū)動
1.1 下載NVIDIA顯卡驅(qū)動程序
從(https://www.geforce.cn/drivers)上選擇支持本機顯卡的的驅(qū)動程序下載,這里選擇NVIDIA-Linux-x86_64-384.98.run藻懒,下載至~/Downloads文件夾

1.2 安裝
首先屏蔽nouveau啟動嬉荆,這里選擇blacklist方式鄙早,將nouveau驅(qū)動添加到黑名單blacklist.conf中

#打開blacklist文件
sudo vi/etc/modprobe.d/blacklist.conf
#在文件末尾添加以下語句
blacklist nouveau
#關閉文件后更新
sudo update-initramfs -u
#重啟系統(tǒng)
reboot

#確認nouveau已被禁止
lsmod | grep nouveau

按Ctrl + Alt + F1到控制臺,關閉當前圖形環(huán)境
輸入用戶名和密碼登錄

#以下三句確保結(jié)束x-window服務
sudo init 3
sudo rm -r /tmp/.X*
sudo service lightdm stop
#進入驅(qū)動下載文件夾
cd ~/Downloads
#改文件權(quán)限為可執(zhí)行文件
chmod +x NVIDIA-Linux-x86_64-384.98.run
#執(zhí)行文件扳缕,--no-opengl-files只安裝驅(qū)動文件,不安裝OpenGL文件
sudo ./NVIDIA-Linux-x86_64-384.98.run --no-opengl-files
#根據(jù)提示完成安裝即可,后重啟
reboot

#檢查驅(qū)動是否安裝成功
nvidia-smi

輸出以下信息說明安裝成功


Fig1.png

2 配置tensorflow-gpu
2.1 安裝CUDA
2.1.1 下載CUDA
https://developer.nvidia.com/cuda-80-ga2-download-archive下載CUDA8.0粥庄。CUDA已更新至9.0,未防止出現(xiàn)不兼容性训堆,建議下載舊版坑鱼。網(wǎng)頁如需注冊鲁沥,可注冊一下画恰,注冊到通過可能需要幾分鐘到十幾分鐘缠局。

2.1.2 安裝

#改變文件權(quán)限
chmod u+x cuda_8.0.61_375.26_linux.run
#執(zhí)行文件
sudo ./cuda_8.0.61_375.26_linux

#接下來進入CUDA的安裝過程甩鳄,按q鍵跳過license說明
#accept接受協(xié)議
#是否安裝NVIDIA啟動選擇否(重要!CUDA自帶的顯卡驅(qū)動兼容性差揖赴,而且我們之前已經(jīng)手動安裝了顯卡驅(qū)動)
Istall NVIDIA AcceleratedGraphics Driver for Linux-X86_64 367.48?
(y)es/(n)o/(q)uit:    #請選擇n

#其他選項按y或默認即可

2.2 安裝cuDNN
2.2.1 下載cuDNN
從(https://developer.nvidia.com/rdp/cudnn-download)下載cuDNN5.1到~/Downloads文件夾燥滑。同樣為防止出現(xiàn)不兼容性,建議下載舊版恃锉,網(wǎng)頁如需注冊肪跋,可注冊一下州既。

2.2.2 安裝

#進入剛剛cuda的安裝目錄
cd /usr/local
#解壓后自動配置
sudo tar -zxvf ~/Downloads/cudnn-8.0-linux-x64-v5.1.tgz
#設置環(huán)境路徑
vi ~/.bashrc
#在文件末尾添加以下三個語句
exportLD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
exportCUDA_HOME=/usr/local/cuda-8.0
exportPATH=/usr/local/cuda-8.0/bin:$PATH
#關閉文件后更新
source ~/.bashrc

#確認完成CUDA安裝
nvcc -V

輸出以下信息說明安裝成功


Fig2.png

2.3 安裝tensorflow-gpu
系統(tǒng):Ubuntu16.04LTS
python:python2.7

tensorflow官網(wǎng)上給出了virtualenv、“native” pip晤郑、Docker磕洪、Anaconda和源代碼安裝五種方式析显,這里推薦使用virtualenv安裝

#安裝pip和virtualenv
sudo apt-get install python-pip python-dev python-virtualenv
#創(chuàng)建virtualenv環(huán)境分尸,命令中“~/tensorflow”可更改為自定義路徑
virtualenv --system-site-packages ~/tensorflow
#激活virtualenv環(huán)境孔庭,之后每次使用tensorflow之前圆到,都需要首先激活環(huán)境
source ~/tensorflow/bin/activate
#命令行改變?yōu)椤?tensorflow)$”
#確保pip版本在8.1以上
easy_install -U pip
#安裝tensorflow-gpu芽淡,tensorflow默認安裝1.4版本,未避免出現(xiàn)不兼容性白胀,這里推薦安裝1.2的較低版本
pip install --upgrade tensorflow-gpu==1.2
#完成安裝纹笼,運行以下程序確認tensorflow-gpu可以運行
python
import tensorflow as tf
hello = tf.constant('Hello,TensorFlow!')
sess = tf.Session()

出現(xiàn)如下信息廷痘,說明tensorflow-gpu啟動成功


Fig3.png
#輸出“Hello, TensorFlow!”說明tensorflow成功運行
print(sess.run(hello))

tensorflow的安裝說明主要來自tensorflow官網(wǎng),如果安裝過程出現(xiàn)問題兄猩,請參閱(https://www.tensorflow.org/install/install_linux)以獲得更多信息和幫助。

鑒于筆者水平有限铜秆,以上安裝步驟如有錯誤或疏漏之處核蘸,煩請不吝指教

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末祟峦,一起剝皮案震驚了整個濱河市宅楞,隨后出現(xiàn)的幾起案子咱筛,更是在濱河造成了極大的恐慌迅箩,老刑警劉巖,帶你破解...
    沈念sama閱讀 211,561評論 6 492
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件奕塑,死亡現(xiàn)場離奇詭異,居然都是意外死亡讨衣,警方通過查閱死者的電腦和手機固蚤,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 90,218評論 3 385
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來燎孟,“玉大人缤弦,你說我怎么就攤上這事碍沐±厶幔” “怎么了磁浇?”我有些...
    開封第一講書人閱讀 157,162評論 0 348
  • 文/不壞的土叔 我叫張陵无虚,是天一觀的道長衍锚。 經(jīng)常有香客問我度宦,道長告匠,這世上最難降的妖魔是什么划鸽? 我笑而不...
    開封第一講書人閱讀 56,470評論 1 283
  • 正文 為了忘掉前任,我火速辦了婚禮建瘫,結(jié)果婚禮上啰脚,老公的妹妹穿的比我還像新娘粒梦。我一直安慰自己荸实,他們只是感情好泄朴,可當我...
    茶點故事閱讀 65,550評論 6 385
  • 文/花漫 我一把揭開白布祖灰。 她就那樣靜靜地躺著局扶,像睡著了一般畜埋。 火紅的嫁衣襯著肌膚如雪悠鞍。 梳的紋絲不亂的頭發(fā)上狞玛,一...
    開封第一講書人閱讀 49,806評論 1 290
  • 那天,我揣著相機與錄音纠吴,去河邊找鬼固该。 笑死糖儡,一個胖子當著我的面吹牛桦沉,可吹牛的內(nèi)容都是我干的纯露。 我是一名探鬼主播代芜,決...
    沈念sama閱讀 38,951評論 3 407
  • 文/蒼蘭香墨 我猛地睜開眼贷掖,長吁一口氣:“原來是場噩夢啊……” “哼羽资!你這毒婦竟也來了?” 一聲冷哼從身側(cè)響起狭郑,我...
    開封第一講書人閱讀 37,712評論 0 266
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎亩鬼,沒想到半個月后雳锋,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體玷过,經(jīng)...
    沈念sama閱讀 44,166評論 1 303
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 36,510評論 2 327
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發(fā)現(xiàn)自己被綠了虑凛。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 38,643評論 1 340
  • 序言:一個原本活蹦亂跳的男人離奇死亡霉囚,死狀恐怖榜跌,靈堂內(nèi)的尸體忽然破棺而出盅粪,到底是詐尸還是另有隱情钓葫,我是刑警寧澤,帶...
    沈念sama閱讀 34,306評論 4 330
  • 正文 年R本政府宣布票顾,位于F島的核電站础浮,受9級特大地震影響,放射性物質(zhì)發(fā)生泄漏奠骄。R本人自食惡果不足惜豆同,卻給世界環(huán)境...
    茶點故事閱讀 39,930評論 3 313
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧鸭廷,春花似錦佛吓、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 30,745評論 0 21
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春苔咪,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背可柿。 一陣腳步聲響...
    開封第一講書人閱讀 31,983評論 1 266
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 46,351評論 2 360
  • 正文 我出身青樓病往,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 43,509評論 2 348

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