1. 程式人生 > >ARM等嵌入式平臺安裝caffe+opencl

ARM等嵌入式平臺安裝caffe+opencl

 博主最近因為專案原因,需要將一個深度學習專案移植到嵌入式開發平臺上,我所使用的是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

        獲取有其他問題,歡迎小夥伴留言討論