DSP

ARM等嵌入式平台安装caffe+opencl

2019-07-13 18:55发布

 博主最近因为项目原因,需要将一个深度学习项目移植到嵌入式开发平台上,我所使用的是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
        获取有其他问题,欢迎小伙伴留言讨论