caffe安装流程及一些问题

最新推荐文章于 2020-05-17 20:39:12 发布

原创最新推荐文章于 2020-05-17 20:39:12 发布 · 402 阅读

0 ·

CC 4.0 BY-SA版权

本文详细记录了在Ubuntu 16.04系统上安装Caffe的过程及其依赖项，包括解决编译过程中遇到的具体问题如hdf5.h文件路径配置错误、内存不足等，并提供了针对不同OpenCV版本的解决方案。

记录一下安装caffe的过程以及其中遇到的问题。ubuntu版本16.04；opencv版本3.2.

1、caffe依赖包安装

sudo apt-get install git

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install python-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

2、下载caffe源码

git clone https://github.com/bvlc/caffe.git
cd caffe
mv Makefile.config.example Makefile.config

在caffe目录ls一下，应该是这样的。

3、编译遇到的问题及如何解决

执行make -j4开始编译。

1、hdf5.h找不到，但是到相应的目录去看，文件就在那里。怎么解决呢？

解决方法
在Makefile.config文件的第85行左右，添加 /usr/include/hdf5/serial/ 到 INCLUDE_DIRS，也就是把下面第一行代码改为第二行代码。
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/

在Makefile文件的第173行左右，把 hdf5_hl 和hdf5修改为hdf5_serial_hl 和 hdf5_serial，也就是把下面第一行代码改为第二行代码。
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

2、在编译这个caffe.pb.cc时，内存不足。解决方法：目前我的虚拟机给了2g内存，关机扩至4g

3、opencv——imgcodecs链接问题

这个问题，如果你的opencv版本为3.0及以上的话，在Makefile.config中将中将OPENCV_VERSION := 3这一句取消注释即可，如下图

若opencv是更早的版本，则要麻烦一些。解决方案是：把opencv需要的lib添加到Makefile文件中，找到LIBRARIES（在PYTHON_LIBRARIES := boost_python python2.7 前一行）并修改为：

LIBRARIES += glog gflags protobuf leveldb snappy \
        lmdb boost_system hdf5_hl hdf5 m \
        opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs