Caffe
是一個清晰而高效的深度學習框架,是純粹的C++、CUDA架構,支持命令行,Python和MATLAB接口,能夠在CPU和GPU直接無縫切換,Caffe的優點:python
上手快,模型和相應優化都是以文本形式而非代碼形式給出,Caffe
給出了模型的定義,最優化設置以及預訓練的權重,方便當即上手。c++
速度快,Caffe與cuDNN結合使用,可以運行最棒的模型和海量的數據。git
模塊化,方便拓展新的認知和設置.github
開源,開放數據庫
Caffe
在MacOS能夠使用Homebrew
安裝,也是自行編譯源代碼安裝,這裏介紹下直接編譯源代碼安裝.bash
Caffe
的github地址:https://github.com/BVLC/caffe,直接克隆Caffe源代碼:架構
git clone git@github.com:BVLC/caffe.gitapp
複製Makefile.config
文件:框架
cd caffe
cp Makefile.config.example Makefile.configpython2.7
Makefile.config
文件中有些編譯選項須要作更改才能在MacOS上面編譯經過,後面會介紹,下面先介紹下編譯Caff所須要的依賴。
編譯Caffe
須要有大量的依賴,開始扁你以前,須要先安裝這些依賴。
(1) 使用GPU模式須要安裝CUDA,安裝CUDA的命令:
brew cask install cuda
也能夠不採用GPU模式,只使用CPU,在Makefile.config
中作修改:
CPUU_ONLY=1
(2) BoostCaffe
使用的是c++開發,若是要使用python調用Caffe
的接口的話,須要安裝boot.python:
brew install boost --with-python
brew install boost-python
(3) OpenCV 安裝,直接安裝OpenCV
brew install opencv
OpenCV安裝以後須要在Makefile.config
中設置OpenCV的文件頭路徑,以及lib的路徑:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/local/Cellar/opencv/2.4.13.2/include LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/local/Cellar/opencv/2.4.13.2/lib
(4) 數據庫leveldb,lmdb,hdf5安裝:
brew install leveldb
brew install lmdb
brew tap homebrew/science
brew install install homebrew/science/hdf5
(5) 日誌與數據操做
brew install protobuf
brew install glog
brew install gflags
brew install snappy
先要安裝 Python依賴庫:numpy,h5py以及scikit-image
brew install numpy
pip install h5py
pip install scikit-image
安裝完Python的依賴類庫以後須要注意如下的4點:
設置Makefile.config
中numpy的路徑:
PYTHON_INCLUDE := /usr/include/python2.7 \
在python中使用OpenCV,須要把OpenCV安裝目錄下../python/site-packages
裏面的兩個文件cv.py
和cv2.so
拷貝到/usr/local/lib/python2.7/site-packages
目錄下,這樣python才能調用OpenCV。
在Makefile.config
設置WITH_PYTHON_LAYER:=1
在Makefile.config
中設置PYTHON_LIB:
PYTHON_LIB := /usr/local/Cellar/python/2.7.13/Frameworks/Python.framework/Versions/2.7/lib
使用make
命令編譯Caffe:
make clean
make all
make test
make runtest
編譯Caffe-Python:
make pycaffe
以上編譯都經過以後將caffe/python添加到python系統路徑裏fish
設置命令:
set -gx PYTHONPATH path/to/caffe/python $PYTHONPATH
bash
的設置命令:
export PYTHONPATH=path/to/caffe/python:$PYTHONPATH
在命令行中直接測試Caffe
是否編譯成功。
python
import caffe
沒有錯誤出現則表示Caffe
安裝成功。