Mac caffe安裝心路

Caffe是一個清晰而高效的深度學習框架,是純粹的C++、CUDA架構,支持命令行,Python和MATLAB接口,可以在CPU和GPU直接無縫切換,Caffe的優勢:

上手快,模型和相應優化都是以文本形式而非代碼形式給出,Caffe給出了模型的定義,最優化設置以及預訓練的權重,方便立即上手。

速度快,Caffe與cuDNN結合使用,能夠運行最棒的模型和海量的數據。

模塊化,方便拓展新的認知和設置.

開源,開放

Caffe在MacOS可以使用Homebrew安裝,也是自行編譯源代碼安裝,這里介紹下直接編譯源代碼安裝.

下載Caffe源代碼

Caffe的github地址:https://github.com/BVLC/caffe,直接克隆Caffe源代碼:

git clone git@github.com:BVLC/caffe.git

復制Makefile.config文件:

cd caffe

cp Makefile.config.example Makefile.config

Makefile.config文件中有些編譯選項需要做更改才能在MacOS上面編譯通過,后面會介紹,下面先介紹下編譯Caff所需要的依賴。

安裝Caffe依賴

編譯Caffe需要有大量的依賴,開始扁你之前,需要先安裝這些依賴。

(1)? 使用GPU模式需要安裝CUDA,安裝CUDA的命令:

brew cask install cuda

也可以不采用GPU模式,只使用CPU,在Makefile.config中做修改:

CPUU_ONLY=1

(2)? Boost

Caffe使用的是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

安裝caffe-python依賴

先要安裝 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 \

/usr/local/Cellar/numpy/1.12.0/lib/python2.7/site-packages/numpy/core/include

在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

編譯Caffe

使用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

在命令行中直接測試Caffe是否編譯成功。

python

import caffe

沒有錯誤出現則表示Caffe安裝成功。

作者:姜家志

鏈接:http://www.lxweimin.com/p/e6d0b11a9e16

來源:簡書

著作權歸作者所有。商業轉載請聯系作者獲得授權,非商業轉載請注明出處。

?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容