Caffe-SSD编译、安装

有些内容可以参考caffe安装https://blog.csdn.net/z_muyangren/article/details/79944717五、安装caffe

1、安装依赖库
之前安装caffe时,已经安装过了,此处不再安装
2、下载源码
github地址为:https://github.com/weiliu89/caffe/tree/ssd
3、修改文件
Makefile.config文件
a、若使用cudnn,则将# USE_CUDNN := 1 修改成: USE_CUDNN := 1
b、若使用python来编写layer,将# WITH_PYTHON_LAYER := 1 修改为 WITH_PYTHON_LAYER := 1
c、*将# Whatever else you find you need goes here.下面的

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include 
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 
修改为:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial 
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial 

//这是因为ubuntu16.04的文件包含位置发生了变化,尤其是需要用到的hdf5的位置
Makefile文件
可以参考caffe的Makefile文件修改,不要直接复制
下图是修改后,与caffe下的Makefile有差别的主要地方的截图:
这里写图片描述
/usr/local/cuda/include/host_config.h

将#/error -- unsupported GNU version! gcc versions later than 5 are not supported!
改为
//#error-- unsupported GNU version! gcc versions later than 5 are not supported!

4、然后进入caffe-ssd目录进行编译

make all –j8(-j8表示并行计算,根据自己的电脑进行配置,或直接输入make all)
make test
make runtest
make pycaffe
make matcaffe(未装matlab,可省略)

同时,sudo gedit ~/.bashrc打开.bashrc文件,在末尾声明Python文件的路径,即写入:

Export PYTHONPATH=/home/.../caffe/python:$PYTHONPATH 

最后,终端输入python,然后再输入import caffe,如果无问题则安装成功。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值