首頁 > 軟體

Ubuntu 16.04.2下Caffe-OpenCL簡要安裝指南

2020-06-16 17:18:21

本人迷信AMD戰未來,於去年畢業後,購買AMD RX480,但是caffe下的GPU加速預設針對n卡,需要安裝cuda。至此要麼換卡,要麼使用opencl版的caffe。想了想,也許rx480以後可以在深度學習領域戰未來呢?於是乎,走上了安裝opencl-caffe的不歸路。

簡要安裝指南:

一、安裝純淨Ubuntu16.04.2系統

這是首先要做的事情,本人使用了win10+ubuntu16.04.2雙系統,為的就是發揮cpu和gpu的所有效能,一波三折,不再贅述。

二、安裝amd顯示卡驅動

顯示卡型號為AMD RX480,驅動amdgpu-pro-16.60-379184.tar.xz,官網下載即可。

三、安裝OpenCL

AMD-APP-SDKInstaller-v3.0.130.136-GA-linux64.tar.bz2

依舊是在AMD官網下載,最後會有一個waiting,時間較長(本來以為是安裝失敗,結果看youtube視訊,跟我最後安裝的結果一樣,才恍然大悟....)

四、下載caffe

Git clone --recursive https://github.com/BVLC/caffe.git

git checkout opencl

mkdir build

cd build

cmake ..

cmake ..  時,會遇到很多沒有安裝的庫,依據錯誤資訊,依次安裝,其中有個比較麻煩ViennaCL,這個庫安裝較為麻煩。 三個問題

(1)會說OPENCL_INCLUDE_DIRS和OPENCL_LIBRARIES找不到,此時要指定這兩個資料夾的路徑

cmake -DOPENCL_INCLUDE_DIRS=/opt/AMDAPPSDK-3.0/include/ -DOPENCL_LIBRARIES=/opt/AMDAPPSDK-3.0/lib/x86_64/libOpenCL.so ..

(2)opencl庫檔案的連結錯誤

AMDAPPSDK-3.0安裝成功後:

/opt/AMDAPPSDK-3.0/lib/x86_64/libOpenCL.so連結有問題,需要手動和/opt/AMDAPPSDK-3.0/lib/x86_64/sdk/libOpenCL.so建立連結

ln -s /opt/AMDAPPSDK-3.0/lib/x86_64/libOpenCL.so /opt/AMDAPPSDK-3.0/lib/x86_64/sdk/libOpenCL.so

(3)

make

make install(需要,後面編譯caffe要用到)

五、編譯Caffe OpenCL

在build資料夾中,緊接著cmake ..

make

大功告成

六、測試

測試藝術遷移

https://github.com/fzliu/style-transfer.git

使用方法可以百度,測試自帶例子,使用gpu,20分鐘左右,使用cpu4個小時左右。

注意:藝術遷移可以使用gpu,但是很多github上的例子都是使用cuda寫的,因此還是有很多問題出現。再就是tensorflow到目前為止,官方還沒有支援opencl,雖然可以在github上下載到其他cl版本的tensorflow,問題較多,不穩定。建議想高效學習深度學習的朋友,直接買n卡。

——————————————

補:還有clBLAS的安裝

Ubuntu 15.04 下Caffe + + CUDA 7.0 安裝設定指南  http://www.linuxidc.com/Linux/2016-11/137497.htm

Caffe 深度學習入門教學  http://www.linuxidc.com/Linux/2016-11/136774.htm

Ubuntu 16.04下Matlab2014a+Anaconda2+OpenCV3.1+Caffe安裝 http://www.linuxidc.com/Linux/2016-07/132860.htm

Ubuntu 16.04系統下CUDA7.5設定Caffe教學 http://www.linuxidc.com/Linux/2016-07/132859.htm

Caffe在Ubuntu 14.04 64bit 下的安裝 http://www.linuxidc.com/Linux/2015-07/120449.htm

深度學習框架Caffe在Ubuntu下編譯安裝  http://www.linuxidc.com/Linux/2016-07/133225.htm

Caffe + Ubuntu 14.04 64bit + CUDA 6.5 設定說明  http://www.linuxidc.com/Linux/2015-04/116444.htm

Ubuntu 16.04上安裝Caffe http://www.linuxidc.com/Linux/2016-08/134585.htm

Caffe設定簡明教學 ( Ubuntu 14.04 / CUDA 7.5 / cuDNN 5.1 / OpenCV 3.1 )  http://www.linuxidc.com/Linux/2016-09/135016.htm

Ubuntu 16.04上安裝Caffe(CPU only)  http://www.linuxidc.com/Linux/2016-09/135034.htm


IT145.com E-mail:sddin#qq.com