[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
英雄XJTU DJI CLUB
2017-4-19 19:53:14 显示全部楼层

马上注册,玩转Robomaster!

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
本文作者:钢蛋,权利归属原作者所有,未获得授权禁止转载!


妙算上面可以安装caffe,已测试成功.写个教程帮助后面想要安装的同学.

百度云链接:http://pan.baidu.com/share/init?shareid=3221400738&uk=1647165732密码:v037

相关安装包在“妙算”文件夹中

1.    首先是cuda的安装.

       妙算中是一块TK1.cuda6.5的离线安装包已上传至百度云,请下载之后,用下列命令安装:

       sudodpkg -i cuda-repo-l4t-r21.3-6-5-prod_6.5-42_armhf.deb

       sudoapt-get update

       sudoapt-get install cuda-toolkit-6-5

添加环境变量

QQ截图20170419200246.jpg

运行下列命令测试是否安装成功


    nvcc -V

1.jpg
                                     

2.    opencv4tegra的安装

       在百度云下载安装包,使用如下命令安装:

       sudodpkg -i libopencv4tegra-repo_l4t-r21_2.4.10.1_armhf.deb

       sudoapt-get update

       sudoapt-get install libopencv4tegra libopencv4tegra-dev libopencv4tegra-python

3.    接下来是caffe需要的各种依赖库,参考caffe官方Installation.


整理如下:

       sudo apt-get install libprotobuf-dev libleveldb-devlibsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

    sudo apt-get install--no-install-recommends libboost-all-dev

   sudoapt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

4.安装openBLAS.这个直接下载下来make,make install,然后需要更新一下ubuntu/etc/ld.so.conf.d中的文件(我在文件夹中添加了一个opencv.conf),openBLASlib路径(默认是/opt/OpenBLAS/lib)写在opencv.conf末尾,然后在终端运行

sudo ldconfig使改动生效

5.解压caffe,然后复制Makefile.config.example并重命名为Makefile.config.打开Makefile.config

1)  找到

  
# CUDA architecture  setting: going with all of them.
  
  
#  For CUDA < 6.0, comment the *_50 through *_61 lines for compatibility.
  
  
# For CUDA < 8.0,  comment the *_60 and *_61 lines for compatibility.
  
  
CUDA_ARCH :=  -gencode arch=compute_20,code=sm_20 \
  
  
-gencode  arch=compute_20,code=sm_21 \
  
  
-gencode arch=compute_30,code=sm_30  \
  
  
-gencode  arch=compute_35,code=sm_35 \
  
  
-gencode  arch=compute_50,code=sm_50 \
  
  
-gencode  arch=compute_52,code=sm_52 \
  
  
-gencode  arch=compute_60,code=sm_60 \
  
  
-gencode arch=compute_61,code=sm_61 \
  
  
-gencode arch=compute_61,code=compute_61
  


然后把最后三行注释掉.变成下面这样:


  
#  CUDA architecture setting: going with all of them.
  
  
# For CUDA < 6.0, comment the *_50 through  *_61 lines for compatibility.
  
  
#  For CUDA < 8.0, comment the *_60 and *_61 lines for compatibility.
  
  
CUDA_ARCH  := -gencode arch=compute_20,code=sm_20 \
  
  
-gencode  arch=compute_20,code=sm_21 \
  
  
-gencode  arch=compute_30,code=sm_30 \
  
  
-gencode  arch=compute_35,code=sm_35 \
  
  
-gencode  arch=compute_50,code=sm_50 \
  
  
-gencode  arch=compute_52,code=sm_52 \
  
  
#-gencode  arch=compute_60,code=sm_60 \
  
  
#-gencode arch=compute_61,code=sm_61  \
  
  
#-gencode  arch=compute_61,code=compute_61
  

2)  修改BLAS的值为open

并修改BLAS_INCLUDEBLAS_LIB的路径

BLAS_INCLUDE:=/opt/OpenBLAS/include

BLAS_LIB:=/opt/OpenBLAS/lib

6.cd caffe的目录下,编译

make all

make test

make runtest


2.jpg
跳转到指定楼层

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
英雄XJTU DJI CLUB
 楼主| 2017-4-19 19:53:48 显示全部楼层
本帖最后由 XJTU DJI CLUB 于 2017-4-20 16:42 编辑

7.(可选)pycaffe的编译。如果想支持python接口,那么需要编译pycaffe。

1) 首先,需要apt-get安装numpy和scipy。

sudoapt-get install python-numpy

sudoapt-get install python-scipy

2) 然后需要用到pip。(没有pip的话先安装pip:sudo apt-get install python-pip)

cd到caffe-master/python目录下,在终端执行下面的命令:

for req in $(cat requirements.txt); do pip install $req; done

如果权限不够,需要sudo,请执行

for req in $(cat requirements.txt); do sudo -H pip install $req;done

3) cd 到caffe-master下,make pycaffe.

4) 更新PYTHONPATH环境变量,把caffe-master/python添加进PYTHONPATH,这样才能在import的时候检索到相应的目录

sudo gedit ~/.bashrc

在文件的末尾添加如下代码:

#add caffe to PYTHONPATH

export PYTHONPATH=$PYTHONPATH:/home/ubuntu/Downloads/caffe-master/python

保存文件之后,在终端执行:

source~/.bashrc

5) 这时候可以在python中import caffe试一下。没有什么提示就是成功了


2.jpg



PS:由于是安装后总结,也许会有疏漏,遇到error请google,baidu,bing...

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
英雄XJTU DJI CLUB
 楼主| 2017-4-19 20:00:06 显示全部楼层
那个……不知道为什么有几句突然成了表情,rio尴尬……晚一点修复

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
工程Snail
2017-4-19 20:01:39 显示全部楼层
XJTU DJI CLUB 发表于 2017-4-19 20:00
那个……不知道为什么有几句突然成了表情,rio尴尬……晚一点修复

感谢分享   

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
工程琪琪心里苦
2017-4-19 20:11:41 显示全部楼层
钢蛋我们是视觉组的dalao~

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
顾问dasuda2015
2017-4-19 20:33:30 显示全部楼层
本帖最后由 dasuda2015 于 2017-4-19 20:45 编辑

厉害!!!之前装过opencv3.1,可以直接用吗,还是需要换回opencv4tegra版本?

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
英雄XJTU DJI CLUB
 楼主| 2017-4-20 17:13:57 显示全部楼层
dasuda2015 发表于 2017-4-19 20:33
厉害!!!之前装过opencv3.1,可以直接用吗,还是需要换回opencv4tegra版本?

我去特地问了大佬XD
QQ截图20170420171315.png

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
梯队队员dugujuan
2018-8-19 11:22:58 显示全部楼层
66666666666666666666666666666666666666666666666666666666

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
顾问梅川酷子
2019-1-19 21:47:25 显示全部楼层
感谢分享 感谢分享

[视觉算法] 【分享帖】教程 || 在秒算上配置caffe

[复制链接]
梯队队员墨雪流年K
2020-8-27 15:37:08 显示全部楼层
感谢大佬分享!
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

触屏版 | 电脑版

Copyright © 2024 RoboMasters 版权所有 备案号 粤ICP备2022092332号

快速回复 返回顶部 返回列表