ARM等嵌入式平臺安裝caffe+opencl
阿新 • • 發佈:2019-01-27
博主最近因為專案原因,需要將一個深度學習專案移植到嵌入式開發平臺上,我所使用的是firefly-rk3399,該專案主要是caffe深度學習框架,之前的程式碼是用cuda進行CPU加速,現在移植到firefly上需要對平臺進行CPU和GPU進行加速,下面開始介紹我安裝caffe+opencl的過程.
安裝caffe
1.首先下載caffe的原始碼,如果沒有git的童鞋可以apt-get 或者pip安裝喲
git clone --recursive https://github.com/BVLC/caffe.git
2.切換到opencl分之,這一步很重要,博主之前就是因為這一步沒做而導致沒安裝成功
git checkout opencl
3.建立一個build檔案,並進入該檔案
sudo mkdir build
cd build
4.進行cmake
sudo cmake -DBLAS=Open ..
5.進行make
sudo make -j4
注意:此時可能會有一些因為缺少依賴的錯誤,你只需要把依賴一一安裝即可,這裡是openBLAS的安裝方法 :https://blog.csdn.net/qq_25680531/article/details/80978473
獲取有其他問題,歡迎小夥伴留言討論