caffe下载安装

1.1 环境搭建 本章所介绍的环境搭建系统为 Ubuntu 14.04 或 Ubuntu 16.04。

1.1.1 Ubuntu 环境设置 本章所介绍的内容可以在 Linux 虚拟机或者 Linux 系统上实现 1.1.2 Caffe 环境设置  下载所需要的依赖库:

在命令行输入以下命令。

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 libgflags-dev libgoogle-glog-dev liblmdb-dev

下载安装 Caffe:

从 GitHub 上下载 Caffe:https://github.com/BVLC/caffe。

也可以输入以 下命令来下载。 sudo git clone https://github.com/BVLC/caffe.git 下载完成后,解压 Caffe 源码包文件。

图 1. 1 Caffe 解压后得到文件

caffe下载安装 

配置 Makefile.config 文件: 解压完成之后,对应文件夹中有 Makefile.config.example 文件,是 Caffe 自带的 Makefile 的例子,复制这个文件并重命名为 Makefile.config,打 开之后

caffe下载安装所示。

图 1. 2 Makefile.config 文件

由于本实例采用 cpu 训练,因此配置的时候需要取消掉 CPU_ONLY := 1 这一行的注释。 

CPU_ONLY := 1

编译 Caffe: 在命令行输入以下命令编译 Caffe。

make all

若 CPU 是 4 核的可以输入(8 核同理)。

make all –j4

如果提示无法找到 hdf5.h 请打开 Makefile.config 更改如下语句 (根据路径找到文件后,修改路径)

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial/

如果提示无法找到 libhdf5.so 请打开 Makefile.config 更改如下语句

LIBRARY_DIRS := $(PYTHON_LIB) /usr/lib/x86_64-linux-gnu/hdf5/serial /usr/local/lib /usr/lib

编译成功后,在 Caffe 根目录下会出现一个 build 文件夹,打开之后再打 开 tools 文件夹,可得到图 1. 3 结果。

caffe下载安装

图 1. 3 编译完成后的/build/tools 文件夹 之后编译 test 文件下的源码,同理输入以下命令即可。

make test

make runtest

 测试: 完成上述操作后即成功安装 Caffe,本实例使用 cifar10 来进行测试,

在 caffe 目 录下依次执行以下操作:

输入以下命令:

./data/cifar10/get_cifar10.sh

./examples/cifar10/create_cifar10.sh

将 examples/cifar10/cifar10_quick_solver.prototxt 中的 GPU 改为 CPU;

将examples/cifar10/cifar10_quick_solver_lr1.prototxt中的GPU改为CPU;

输入命令:

./examples/cifar10/train_quick.sh

训练不久会出现如图 1. 4 所示的画面,即表明配置成功。 图 1. 4

caffe下载安装

训练过程截图

1.1.3

pycaffe 环境设置 Caffe 训练及测试时会用到 python 接口,下面介绍 pycaffe 接口的配置,其流程如下:

下载库:

sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python3-pip python-pip

sudo pip install python-skimage

sudo apt-get install python-h5py python-protobuf python-leveldb python-networkx pythonnose python-pandas python-gflags sudo pip install Cython ipython 

编译: 在命令行输入以下命令,即可开始编译 python 接口。

cd caffe make pycaffe

添加环境变量: 在/etc/profile 文件的最后一行中添加环境变量:

export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH 之后

输入以下命令,使之生效。

source /etc/profile

 测试: 在命令行输入:

cd python python

在 python 中输入:

import caffe 若结果如图 1. 5 所示,则 python 接口配置成功。

caffe下载安装

图 1. 5 python 接口配置结果 若出现”No module named _caffe”的错误,可尝试重新输入以下命令,

make clean

make pycaffe