Ubuntu18.04 安装SSD-Caffe

• 环境背景:

由于项目需要，在ubuntu 18.04 下安装Caffe， 前期安装了 CUDA9.0 ，GPU为 NVIDIA GTX 1080。

• 安装caffe参考：https://blog.csdn.net/CAU_Ayao/article/details/84000151
• 系统版本：Ubuntu18.04
• 系统环境：OpenCV2.4.13.4，Python2
• OpenCV2.4.13.4安装:ubuntu 18.04 安装opencv2.4.13.4 遇到的问题及解决方法

1. 安装依赖项

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev
sudo apt-get install 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. Clone Caffe-SSD源码

git clone https://github.com/weiliu89/caffe.git caffe-ssd
cd caffe
git checkout ssd

这里需出现分支成功的提示方可继续：

Branch ssd set up to track remote branch ssd from origin.
Switched to a new branch ‘ssd’

3. 文件配置

cd caffe-ssd
cp Makefile.config.example Makefile.config

  修改Makefile.config文件

1.若使用Cudnn
  将# USE_CUDNN := 1 前的注释取消
  即：USE_CUDNN := 1

2.若使用opencv3.x
  取消 # OPENCV_VERSION := 3 前的注释
  即：OPENCV_VERSION := 2
 
3.若是用Python编程
  取消 # WITH_PYTHON_LAYER := 1 前的注释
  即：WITH_PYTHON_LAYER := 1
 
4.加入hdf5的目录：
  将
    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

5. 如果使用的是Cuda>8.0，则将
    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_61,code=sm_61
    修改为：
    CUDA_ARCH := -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_61,code=sm_61

  

    修改Makefile文件

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

 

NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
 
修改为：
 
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

   

LIBRARIES += boost_thread stdc++后加boost_regex
 
修改为:
 
LIBRARIES += boost_thread stdc++ boost_regex

  

4. 编译Caffe-SSD

sudo make -j8
sudo make py
sudo make runtest -j8
sudo make test -j8

5. import caffe error

这样都很成功的编译完了，以为万事ok了，可是在进入caffe/python，打开python之后，import caffe却出现了

python caffe libcaffe.so.1.0.0-rc3: cannot open shared object file错误

from ._caffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, \
ImportError: libcaffe.so.1.0.0-rc3: cannot open shared object file: No such file or directory

参考：https://blog.csdn.net/thesby/article/details/50791469#commentBox

造成这个问题的主要原因在于使用make进行编译,导致import caffe时找不到静态链接库.

现给出两种比较安全的解决方案:
(1) 在python 代码中加入如下代码:

import os, sys
old_path = os.getcwd()
caffe_root = "caffe-master"
os.chdir(caffe_root + "/python")
import caffe
os.chdir(old_path)

这部分代码就是先切换到caffe的python目录,然后导入caffe库,然后在切换回原来目录.

实验了这种方法没有成功，然后就又删除了重新编译的。

(2) 其实找不到libcaffe.so.1.0.0-rc3主要原因在于使用了make进行编译,使得caffe的python接口只能在这个目录下面才能找得到这个库.解决办法是使用cmake进行编译:

cd caffe-master
mkdir build
cd build
cmake ..

编译之前，先按照这个步骤cmake，之后再去执行前面的第4步骤，编译成功。开心。。。。。

 

之后再import caffe测试，又出现了以下两个小问题，很好解决，ok完工。

• 遇到的问题及解决 方案：

1. 出现ImportError: No moudle named skimage.io错误：

解决方法：

sudo pip install scikit-image

sudo apt-get install python-skimage

2. 出现 ImportError: No moudle named google.protobuf.internal 错误：

解决方法：

pip install protobuf