ubuntu16.04安裝caffe的GPU版本


即使找到了安裝方法,每個人的系統多少都有些差異,總有些坑要踩過才知道實際情況是怎么樣的。我的環境是Lenovo V480 + Ubuntu 16.04 + GeForce GT 645M。安裝過程是參考這篇博客——Ubuntu 16.04 安裝配置Caffe 圖文詳解。需要完成的步驟有:

  1. 安裝相關依賴項
  2. 安裝NVIDIA驅動
  3. 安裝CUDA
  4. 配置cuDNN
  5. 安裝OpenCV
  6. 安裝Caffe
  7. 測試

這位博主的過程記錄很詳細,但有些細節方面我的問題有些不一樣。我記錄我的問題,希望有同樣問題的人可以得到幫助

  1. NVIDIA顯卡驅動安裝
  2. 安裝OpenCV
  3. 安裝caffe
  4. 后記

1.NVIDIA顯卡驅動安裝

開始安裝NVIDIA顯卡驅動進入一個“building kernel”的界面,中間的進度條卡在100%好長時間,我以為已經安裝完成就強制關機了。但開機發現筆記本變得超級慢,查詢顯卡驅動也顯示未定義命令。這才意識到顯卡驅動並沒有安裝完成。實際整個安裝過程的確很耗時,而且也沒什么提示,所以不要關機,耐心等他裝完。中間還給出兩次警告,一次是提示“pre-install script failed........”,我選擇繼續安裝。還有一次是提示什么配置,我也是選擇繼續。安裝結束后提示“Unable to find a suitable destination to install 32-bit compatibility”。我不是很能理解為什么安裝64位驅動會涉及與32位的匹配問題。
第一遍裝完后系統還是有問題,一方面分辨率不對,另一方面循環進入登錄界面。解決方案是參考這篇博客Ubuntu安裝NVIDIA驅動后桌面循環登錄問題。先刪除了NVIDIA的驅動文件,然后用

sudo ./NVIDIA-Linux-x86_64-375.20.run –no-x-check –no-nouveau-check –no-opengl-files

當然驅動文件名是自己的。這一次在tty1終端中,安裝的時候並沒有停止X-Windows服務,發現對安裝並沒有影響。這一次驅動安裝成功,但是分辨率還是有問題,借鑒這篇博客 ubuntu16.04中用xrandr設置屏幕分辨率,且重啟后不失效。到這里顯卡驅動終於修正過來了。

不過輸入“nvidia-smi”彈出的信息顯示GPU的Process name項為Not Supported。我瀏覽其他人的看法。有人認為這是NVIDIA對老顯卡的顯示信息不正確,但實際上顯卡也是工作的。也有人認為是新的驅動與顯卡兼容性不好,所以顯卡不能正常工作,導致實際上是CPU中的集顯在承擔任務。但無論怎樣並不影響后續的安裝與caffe的運行。

2.安裝OpenCV

我筆記本之前已經安裝了OpenCV,本以為可以不用安裝,所以直接進行caffe的安裝,結果出問題了。所以把OpenCV卸載之后重新編譯安裝一遍。結果超出想象。原本OpenCV只用一個小時左右就可以安裝完成,沒想到在CUDA的環境下用了5個多小時才編譯完成。或許是我的電腦太渣了。所以OpenCV一定要在CUDA環境配置好了之后再安裝。

安裝OpenCV中反復出現一個warning,not used because“ OPENCV_TRAITS_ENABLE_DEPRECATED ”has been defined。幾乎每編譯一個文件都會彈出這個警告。實踐證明,這個警告並不影響最終的程序執行。只要沒有error,warning一般可以忽略。

3.caffe的安裝

到這一步是最大的坑了。之前的步驟按照博主的教程,雖然有點差異,但方向都沒有問題,自己稍微做一點修正就可以了。但是caffe的編譯博主犯了錯。

一般我們使用源文件安裝的方式都是使用如下步驟的

mkdir build
cd build
cmake ..
make

但是博主對一些文件設置完成直接用make all -j8。我當時也沒多想,按照命令往下做。但無論怎么修改都會出現nvcc fatal: Unsupported gpu architecture 'compute_20'的錯誤提示。改了3次,才意識到好像還沒做cmake。這樣才安裝成功。

4.后記

在測試的時候發現速度的確很慢,有可能是我的筆記本太渣了,也有可能是顯卡驅動和顯卡不適配,所以GPU加速沒有起作用。不管怎樣這都是第一步。現在先用起來,以后對計算有要求了可以加載到雲上去,或者買更好的計算資源。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM