Ubuntu16.04 Caffe 安装步骤记录

我就是我 2022-04-04 06:12 255阅读 0赞

环境：  
Cuda: 8.0.61  
Cudnn: 6.0.21  
前期准备啥的就不赘述了，下面正式开始安装caffe

--------------------

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

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

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

sudo cp Makefile.config.example Makefile.config

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

sudo gedit Makefile.config

4 修改 Makefile.config 文件内容：  
a 应用 cudnn

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

b 应用 opencv 版本

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

c 使用 python 接口

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

d 修改 python 路径

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 然后修改 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

6 然后修改 /usr/local/cuda/include/host\_config.h 文件 :

将
    #error-- unsupported GNU version! gcc versions later than 4.9 are not supported!
    改为
    //#error-- unsupported GNU version! gcc versions later than 4.9 are not supported!

7 OK ，可以开始编译了，在 caffe 目录下执行 ：

make all -j8

8 安装 pycaffe notebook 接口环境  
a 编译 pycaffe :

cd caffe
    
    sudo make pycaffe -j8

9 最重要的是：要添加环境变量哦

vim ~/.bashrc  # 最后一行添加
    export PYTHONPATH=/home/xxxx/Documents/caffe/python:$PYTHONPATH

然后再重新加载：`source ~/.bashrc`