深度学习框架Caffe的环境搭建
前言：完全小白，一步步摸索，限于硬件条件（没有Nvidia显卡），本笔记只搭建CPU模式（以后有条件再整GPU版本）。
操作系统：Ubuntu 16.04
编程接口：Python 2.7
第一部分：安装依赖包
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
apt-get
install
libprotobuf-dev
apt-get
install
libleveldb-dev
apt-get
install
libsnappy-dev
apt-get
install
libopencv-dev
apt-get
install
libhdf5-serial-dev
apt-get
install
protobuf-compiler
apt-get
install
--no-
install
-recommends libboost-all-dev
apt-get
install
libgflags-dev
apt-get
install
libgoogle-glog-dev
apt-get
install
liblmdb-dev
apt-get
install
libatlas-base-dev
apt-get
install
libboost-all-dev
apt-get
install
python-dev
|
PS：每一个依赖包的什么作用详见《21天实战Caffe》
第二部分：下载Caffe
① 用git命令直接将caffe下载下来，没有git的先安装git工具
|
1
2
|
apt-get
install
git
git clone https:
//github
.com
/BVLC/caffe
.git
|
② 下载完成后切换到Caffe所在目录，然后生成自己的配置文件Makefile.config
|
1
|
cp
Makefile.config.example Makefile.config
|
③ 接着用用vim或者gedit打开修改配置文件Makefile.config
要做两件事：
1)仅CPU模式开启
|
1
|
CPU_ONLY := 1
|
2)配置一些引用文件（增加部分主要是解决新版本下，HDF5的路径问题）
|
1
2
3
4
5
6
|
INCLUDE_DIRS := $(PYTHON_INCLUDE)
/usr/local/include
/usr/lib/x86_64-linux-gnu/hdf5/serial/include
LIBRARY_DIRS := $(PYTHON_LIB)
/usr/local/lib
/usr/lib
/usr/lib/x86_64-linux-gnu/hdf5/serial3
)BLAS := atlas
|
第三部分：编译Caffe
开始编译Caffe
|
1
2
3
|
make
all
make
test
make
runtest
|
第四部分：编译Python接口
Caffe拥有python\C++\shell接口，在Caffe使用python特别方便，在实例中都有接口的说明。
① 首先确保pip已经安装，并且是最新版的。
用apt-get安装pip
|
1
|
sudo
apt-get
install
python-pip
|
但是有时通过apt-get安装的pip版本太老了，使用旧版本pip安装一些包时会报出提醒来升级pip。如果想升级最新的pip，需要先卸载pip，命令为
|
1
|
sudo
apt-get remove python-pip
|
然后去Python官网https://pypi.python.org/pypi/pip 这个网站下载对应版本的源码：
推荐为这种压缩格式的：pip-9.0.1.tar.gz，然后到下载目录下，解压并进入文件：
|
1
2
|
tar
-xzvf pip-9.0.1.
tar
.gz
cd
pip-9.0.1
|
然后手动安装
|
1
|
sudo
python setup.py
install
|
然后执行如下，安装成功。
|
1
2
3
4
|
running
install
... ...
Processing dependencies
for
pip==9.0.1
Finished processing dependencies
for
pip==9.0.1
|
好了，至此pip已经安装好了，之所以安装pip，是为了更好的管理Python包。
② 执行安装依赖
在caffe根目录的python文件夹下，有一个requirements.txt的清单文件，上面列出了需要的依赖库，按照这个清单安装就可以了。
在安装scipy库的时候，需要fortran编译器（gfortran)，如果没有这个编译器就会报错，因此，我们可以先安装一下。
首先回到caffe的根目录，然后执行安装代码：
|
1
2
3
|
cd
/caffe
sudo
apt-get
install
gfortran
cd
.
/python
|
然后执行以下shell脚本命令：
|
1
|
for
req
in
$(
cat
requirements.txt);
do
pip
install
$req;
done
|
安装完成以后，再次回到caffe根目录我们可以执行：
|
1
|
sudo
pip
install
-r python
/requirements
.txt
|
安装成功的，会显示Requirement already satisfied。
③ 编译python接口
|
1
|
make
pycaffe
|
结果显示ALL TESTS PASSED 就说明安装好了！
④ 运行python结构
|
1
2
3
4
5
6
|
python2.7
Python 2.7.12 (default, Jul 1 2016, 15:12:24)
[GCC 5.4.0 20160609] on linux2
Type
"help"
,
"copyright"
,
"credits"
or
"license"
for
more
information.
>>>
import
caffe
>>>
|
如果没有报错，说明caffe安装全部完成(注意：要进入caffe/python再执行python命令，否则import caffe会提示找不到caffe)！
到这里，整个caffe环境搭建就完成了！
接下来就要跑一个小例子来验证下了。
Mnist手写体识别，号称是深度学习界的helloworld，哈哈哈哈。
参考资料：
1、http://www.cnblogs.com/xuanxufeng/p/6016945.html
2、http://www.linuxidc.com/Linux/2016-09/135034.htm
3、http://www.linuxdiyf.com/linux/26860.html