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/zhoudeliang/caffe/python:$PYTHONPATH 

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

猜你喜欢

转载自blog.csdn.net/z_muyangren/article/details/80300923