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

        获取有其他问题,欢迎小伙伴留言讨论

猜你喜欢

转载自blog.csdn.net/qq_25680531/article/details/80978183