Mac 下配置 Caffe（CPU Only）

发表于   |   分类于   |     |   阅读次数

安装依赖

首先用 Homebrew 安装（或重装）依赖项：

1
2
3
4
5
6
7
for x in snappy leveldb gflags glog szip hdf5 lmdb homebrew/science/opencv;
do
    brew uninstall $x;
    brew install --fresh -vd $x;
done
brew uninstall --force protobuf; brew install --with-python --fresh -vd protobuf
brew uninstall boost boost-python; brew install --fresh -vd boost boost-python

注意，如果提示 error 说 brew link xxx 不成功的，往往是指定路径下有以前的包文件，发生冲突。按照提示的路径删除旧文件，重新 link 即可。

安装 Caffe

下载 caffe 到合适的地方（配置完成后使用时不能移动或删除，所以首先找好地方放置），并修改配置：

1
2
3
git clone https://github.com/BVLC/caffe.git
cd caffe
cp Makefile.config.example Makefile.config

打开 Makefile.config，取消 CPU_ONLY:=1 的注释。打开 CMakeLists.txt，找到 CPU_ONLY 一项，把 OFF 改成 ON。

配置 Makefile：

1
2
3
mkdir build
cd build
cmake ..

安装：

1
2
3
4
make all
make runtest
make pycaffe            # Python 接口
make pytest             # Python 测试代码

如果需要 caffe 的 python 支持，则把路径 caffe/python 加入到 PYTHONPATH 中去，打开 ~/.bash_profile，在最后添加：

1
PYTHONPATH=/CAFFE_HOME/caffe/python:$PYTHONPATH

然后 source ~/.bash_profile 使其生效。

测试

首先测试 C++ 版本，运行 Mnist 例子。cd 到 caffe 主目录下：

1
2
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh

运行不出错，说明安装没问题。

再试试 python。打开 python，import caffe 看看。如果不出错，说明配置成功。

可以把 build/lib 目录下的文件复制到 /usr/local/lib 下方便使用：

1
cp -a build/lib/. /usr/local/lib/

也可以把 include/caffe 目录放到 /usr/local/include 中去，方便写程序时查找和配置。

参考链接

[1] MAC OS X10.10下Caffe无脑安装（CPU ONLY)

[2] 深度学习框架Caffe在Mac上的安装和测试