1. 程式人生 > >深度學習準備之Ubuntu下裝顯示卡驅動

深度學習準備之Ubuntu下裝顯示卡驅動

去年下半年看了相關目標檢測的論文,一些傳統的演算法,一些CVPR,TPAMI,ECCV,ICCV,,NIPS,比較前沿的進展,主要都是基於深度學習卷積神經網路方面的,包括RCNN,SPP-NET,Fast-RCNN,Faster-RCNN,以及YOLO,SSD,然後明天準備正式開始搞學術了,爭取早點把論文發出來。

趁著放假回家的時間,做點前期準備工作,深度學習方面的框架有caffe,torch,mxnet,kerse,還有最近比較大熱的TensorFlow,感覺tf會持續popular,所以準備搭建一個玩玩,但是相關比較新的目標檢測框架都是首先基於caffe開發的,caffe對於影象方面的庫比較多,穩定,所以搭建一個caffe也是必不可少,好像tf也是可以呼叫caffe框架模型的(沒用過)。

裝顯示卡驅動之step two(很多同學都是因為這步而放棄深度學習的):

這也是使用深度學習框架GPU版本的關鍵,無奈裝了幾天沒裝好,大部分時間都是停在這一步,只能放棄,其實這裡與我的電腦顯示卡設定有關,後面再說,大部分人還是可以安裝成功的,最後我在雙系統Ubuntu16.04中安裝了caffe和TensorFlow的cpu版本,但是不甘心啊,還好後面在Windows下安裝caffe的GPU版本異常的順利,竊喜。

我的顯示卡:


我也是反覆安裝了兩三天才發現這個問題,浪費了我很多時間啊,當然你如果lspci |grep VGA是兩個都是顯示VGA的,你可以安裝顯示卡驅動的,這裡主要有三種方法。

一是:系統設定->軟體更新->附加驅動->選擇nvidia最新驅動(361)->應用更改

簡單但是不提倡

二是:先官網下載好對應驅動編譯

1)開啟終端,先刪除舊的驅動:

sudo apt-get purge nvidia*

2)禁用自帶的 nouveau nvidia驅動

建立一個檔案通過命令 sudo vim /etc/modprobe.d/blacklist-nouveau.conf

並新增如下內容:

blacklist nouveau
options nouveau modeset=0

再更新一下

sudo update-initramfs -u

修改後需要重啟系統。確認下Nouveau是已經被你幹掉,使用命令: lsmod | grep nouveau

3)重啟系統至init 3(文字模式),也可先進入圖形桌面再執行init 3進入文字模式,再安裝下載的驅動就無問題,

首先我們需要結束x-window的服務,否則驅動將無法正常安裝

關閉X-Window,很簡單:sudo service lightdm stop,然後切換tty1控制檯:Ctrl+Alt+F1即可

4)接下來就是最關鍵的一步了:sudo./NVIDIA.run開始安裝,安裝過程比較快,根據提示選擇即可最後安裝完畢後,重新啟動X-Window:sudo service lightdm start,然後Ctrl+Alt+F7進入圖形介面;


最後測試一下是否安裝成功

nvidia-smi

nvidia-settings

三是:新增官方ppa源

快捷鍵ctrl+alt+T開啟命令終端,加入官方ppa源。

$ sudo add-apt-repository ppa:graphics-drivers/ppa

需要輸入密碼並按enter鍵確認。之後重新整理軟體庫並安裝最新驅動。

$ sudo apt-get update

$ sudo apt-get install nvidia-367nvidia-settings nvidia-prime

安裝完成後通過下面命令檢視是否安裝成功。

$ nvidia-settings

注意安裝完成後要重啟,有如下效果則安裝完成,否則就說明安裝有問題,嘗試關閉UEFI保護試試。