前言:caffe官网中安装caffe时涉及的东西比较多而杂乱,其实有些东西可以忽略掉。
步骤:
mac10.12本身自带了python2.7,可以使用系统自带的python
1.安装Homebrew,pip,virtualenv(用于创建基于不同python版本的虚拟python环境,隔离不同python项目依赖的python环境)
在终端中输入 ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
pip ,virtualenv的安装自行google
2.安装caffe需要用到的一些依赖库,在终端中依次输入下面几行
brew install -vd snappy leveldb gflags glog szip lmdb
# need the homebrew science source for OpenCV and hdf5
brew tap homebrew/science
brew install hdf5 opencv
# with Python pycaffe needs dependencies built from source
brew install --build-from-source --with-python -vd protobuf
brew install --build-from-source -vd boost boost-python
3.安装caffe需要的BLAS,caffe官网描述(如下)的是可以选择使用mac自带的 Accelerate / vecLib Framework,也可以选择OpenBLAS和MKL。我没整明白Accelerate / vecLib Framework,我选择的是安装 openblas。
BLAS: already installed as the Accelerate / vecLib Framework. OpenBLAS and MKL are alternatives for faster CPU computation.
下载 openblas源码并编译安装到/usr/local目录:
tar xvf OpenBLAS-0.2.19.tar.gz
cd OpenBLAS-0.2.19
make
make PREFIX=/usr/local install
到这里,caffe需要依赖的所有库安装完毕,开始安装caffe
4.安装caffe
4.1.创建虚拟的python环境:
在终端执行:
virtualenv --no-site-packages vpy2.7forcaffe -p /System/Library/Frameworks/Python.framework/Versions/2.7/bin/python2.7
source vpy2.7forcaffe/bin/activate #进入python的虚拟环境vpy2.7forcaffe
4.2.拉取caffe的源码
在终端执行以下几行:
git clone https://github.com/BVLC/caffe
cd /path/to/caffe #caffe的下载地址
cp Makefile.config.example Makefile.config
4.3.安装caffe中的python需求环境(进行基于caffe的python开发所需要的python库)
在终端执行以下几行:
cd /path/to/caffe/python
for req in $(cat requirements.txt); do pip install $req; done
4.4.修改Makefile.config文件
1. 将 CPU_ONLY := 1的注释去掉,这样Caffe就只能在CPU上运行了
2.更改BLAS的配置
# BLAS choice:
# atlas for ATLAS (default)
# mkl for MKL
# open for OpenBlas
BLAS := open
# Custom (MKL/ATLAS/OpenBLAS) include and lib directories.
# Leave commented to accept the defaults for your choice of BLAS
# (which should work)!
BLAS_INCLUDE := /usr/local/include
BLAS_LIB := /usr/local/lib
然后在终端执行:
make all
make test
make runtest
caffe安装成功的话,会显示如下:
5.为了使用caffe的python接口,需要安装pycaffe
在执行make pycaffe
前,修改MakeFile.config文件的相关配置,如下:
修改1:
PYTHON_INCLUDE变量:将前面创建的虚拟python环境vpy2.7forcaffe中对应的include路径追加到PYTHON_INCLUDE
PYTHON_LIB变量:将前面创建的虚拟python环境vpy2.7forcaffe中对应的lib路径追加到PYTHON_LIB
修改2:numpy库的的inclubde路径和lib路径分别追加到PYTHON_INCLUDE和PYTHONE_LIB变量中:
然后在终端执行如下操作:
cd /Users/XXX/caffe #caffe的下载地址
make pycaffe