Ubuntu Anaconda3 环境下安装caffe

2022-11-11 08:38:20

安装Python环境

本人环境为Anaconda3 ，可参照 https://blog.csdn.net/ctwy291314/article/details/86571198 完成安装Python2.7环境安装与多Python环境切换。

安装依赖包

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 libopenblas-dev liblapack-dev libatlas-base-dev

sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

sudo apt-get install git cmake build-essential

安装显卡驱动与CUDA 、cuDNN

请参照 https://blog.csdn.net/ctwy291314/article/details/80951148 进行CUDA 、cuDNN进行安装。

安装protoc

命令 whereis protoc可以查看哪些路径下安装了protoc
命令which protoc可以查看默认选用protoc的路径
命令 protoc --version可以查看当前protoc版本

安装protobuf2.6.1

下载文件
https://github.com/google/protobuf/releases/download/v2.6.1/protobuf-2.6.1.tar.gz
解压安装

tar -zxvf protobuf-2.6.1.tar.gz
sudo apt-get install build-essential
cd protobuf-2.6.1/
./configure
make
make check
sudo make install

指定protoc的版本

在Makefile 中修改这两句：

$(Q)protoc --proto_path=$(PROTO_SRC_DIR) --cpp_out=$(PROTO_BUILD_DIR) $<
$(Q)protoc --proto_path=src --python_out=python $<
为
$(Q)/usr/bin/protoc --proto_path=$(PROTO_SRC_DIR) --cpp_out=$(PROTO_BUILD_DIR) $<
$(Q)/usr/bin/protoc --proto_path=src --python_out=python $<

即把开头的"protoc"补全路径即可 (/usr/bin/protoc即为自己向指定给的版本路径)
注：这种修改不会影响系统默认的protoc版本，只会在caffe编译的时候调用相应的proto版本

安装 caffe

首先在你要安装的路径下 clone ：

git clone https://github.com/BVLC/caffe.git

进入 caffe ，将 Makefile.config.example 文件复制一份并更名为 Makefile.config ，也可以在 caffe 目录下直接调用以下命令完成复制操作：

cd caffe
sudo cp Makefile.config.example Makefile.config

复制一份的原因是编译 caffe 时需要的是 Makefile.config 文件，而Makefile.config.example 只是caffe 给出的配置文件例子，不能用来编译 caffe。

然后修改 Makefile.config 文件，在 caffe 目录下打开该文件：

sudo gedit Makefile.config

修改 Makefile.config 文件内容：

应用 cudnn

    #将
    #USE_CUDNN := 1
    #修改成： 
    USE_CUDNN := 1

应用 opencv 版本

    #将
    #OPENCV_VERSION := 3 
    #修改为： 
    OPENCV_VERSION := 3

使用 python 接口

	#将
	#WITH_PYTHON_LAYER := 1 
	 #修改为 
	WITH_PYTHON_LAYER := 1

修改 python 路径

修改PYTHON_INCLUDE、PYTHON_LIB路径
本人安装的是Anaconda3多Python版本切换，python2.7 目录为：/home/hylink/anaconda3/envs/python27
Ubuntu Anaconda3 环境下安装caffe
再修改INCLUDE_DIRS、LIBRARY_DIRS 路径

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

修改 caffe 目录下的 Makefile 文件

#将：
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
#替换为：
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

#将：
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

开始编译
make all -j8

编译出现问题1：

NVCC src/caffe/layers/bnll_layer.cu  
nvcc fatal   : Unsupported gpu architecture 'compute_20'  
Makefile:594: recipe for target '.build_release/cuda/src/caffe/layers/bnll_layer.o' failed  
make: *** [.build_release/cuda/src/caffe/layers/bnll_layer.o] Error 1  
make: *** Waiting for unfinished jobs....

仔细查看了一下 Makefile.config 中 CUDA_ARCH 设置未按规定设置：

# CUDA architecture setting: going with all of them.  
# For CUDA < 6.0, comment the *_50 through *_61 lines for compatibility.  
# For CUDA < 8.0, comment the *_60 and *_61 lines for compatibility.  
# For CUDA >= 9.0, comment the *_20 and *_21 lines for compatibility.  
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_60,code=sm_60 \  
            -gencode arch=compute_61,code=sm_61 \  
            -gencode arch=compute_61,code=compute_61

因为我装的是CUDA9.0所以把下面这两行删除就可以了

-gencode arch=compute_20,code=sm_20 \  
-gencode arch=compute_20,code=sm_21 \

编译出现问题2：

error: #error -- unsupported GNU version! gcc versions later than 6 are not supported!
错误的解决方式很简单，就是安装低版本gcc和g++，并创建链接或更改gcc各版本的优先级。
此处为以后考虑安装了gcc-6和g+±6,大家可以自行决定版本，只要比错误中提到的支持上限小就没问题

sudo apt-get install gcc-6
sudo apt-get install g++-6

然后执行

sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-6 10
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-6 10

编译出现问题3：

error while loading shared libraries: libcudart.so.9.0: cannot open shared object file: No such file
error while loading shared libraries: libcudnn.so.7: cannot open shared object file: No such file

解决方法

sudo cp /usr/local/cuda-9.0/lib64/libcudart.so.9.0 /usr/local/lib/libcudart.so.9.0 && sudo ldconfig
sudo cp /usr/local/cuda-9.0/lib64/libcublas.so.9.0 /usr/local/lib/libcublas.so.9.0 && sudo ldconfig
sudo cp /usr/local/cuda-9.0/lib64/libcurand.so.9.0 /usr/local/lib/libcurand.so.9.0 && sudo 		ldconfig
sudo cp /usr/local/cuda-9.0/lib64/libcudnn.so.7 /usr/local/lib/libcudnn.so.7 && sudo ldconfig

编译成功后可运行测试：
sudo make runtest -j8

安装 pycaffe

首先编译 pycaffe :

cd caffe
sudo make pycaffe -j8

编译 pycaffe 成功后，验证一下是否可以在 python 中导入 caffe 包，首先进入 python 环境：

python

然后导入 caffe :

>>> import caffe

错误1：
ModuleNotFoundError: No module named 'caffe._caffe'

码农公寓