- 环境背景:
由于项目需要,在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