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 解压后得到文件
配置 Makefile.config 文件: 解压完成之后,对应文件夹中有 Makefile.config.example 文件,是 Caffe 自带的 Makefile 的例子,复制这个文件并重命名为 Makefile.config,打 开之后
图 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 结果。
图 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
训练过程截图
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 接口配置成功。
图 1. 5 python 接口配置结果 若出现”No module named _caffe”的错误,可尝试重新输入以下命令,
make clean
make pycaffe