
Mac OS 安裝 Caffe2 且支援GPU 的步驟
新版 PyTorch 已內含 Caffe2,所以不需要再額外安裝 Caffe2,請直接安裝 PyTorch (更新:2018/4)
以下是舊版本安裝,或是單獨安裝Caffe2的方法:
Step 0: (重要:必需先安裝好Cuda及Cudnn)
安裝步驟請看另一篇文章:Mac OS 安裝Cuda及Cudnn
Step 1: 參考官網MAC安裝說明,用brew安裝所有的相依套件:
brew install ffmpeg …
brew install opencv@2
Step 2: 新版 Caffe2 已支援 Python3,不需要再安裝Python2 (更新:2018/4)
Install Python 2 (舊版Caffe2只支援Python 2)
brew install python2
如果已經用 Anaconda 安裝 python2,請切換到 python2
source activate python2
Step 3: brew install, pip install
brew install graphviz
pip install future numpy protobuf protobuf six scikit-image graphviz pydot
pip install --upgrade git+https://github.com/Maratyszcza/PeachPy
pip install --upgrade git+https://github.com/Maratyszcza/confu
Step 4:
Download 『caffe2』
git clone --recursive https://github.com/caffe2/caffe2.git
最新版Caffe2已經併入PyTorch,可以下載PyTorch再進入caffe2目錄 (更新:2018/4)
Step 5: Patch ./caffe2/CMakeLists.txt, Add:
cmake_minimum_required(VERSION 3.10)
set(CMAKE_MACOSX_RPATH 0)
# 如果沒加上這一行,會出現『Policy CMP0042 is not set』錯誤
Step 6: 新版不需要修正
Patch ./caffe2/binaries/CMakeLists.txt, Delete:
#caffe2_binary_target("speed_benchmark.cc")
Step 7: 新版不需要修正
Patch ./caffe2/operators/tile_op.h, Add:
#include <array>
Step 8: Set Environment Variables:
export CUDA_HOME=/Developer/NVIDIA/CUDA-8.0
export DYLD_LIBRARY_PATH="$DYLD_LIBRARY_PATH:$CUDA_HOME/lib"
export PATH="$CUDA_HOME/bin:$CUDA_HOME:$PATH"
export CMAKE_LIBRARY_PATH=$CMAKE_LIBRARY_PATH:$HOME/caffe2/third_party/nccl/build/lib
Step 9: 設定使用 Xcode_8.2.1 編譯 (注意:大部分Xcode都不支援 Cuda編譯)
sudo xcode-select -s /Applications/Xcode_8.2.1/Xcode.app/Contents/Developer/sudo xcode-select -p
# 確定是否已切換為 Xcode_8.2.1
# 必需使用 『Xcode_8.2.1』,否則會出現類似下面的錯誤訊息:
# The version (‘90000’) of the host compiler (‘Apple clang’) is not supported
Step 10:
cd caffe2
mkdir build
cd build
Step 11:
cmake -DBUILD_TEST=OFF -DUSE_FFMPEG=ON .. # with GPU#
cmake -DUSE_CUDA=OFF .. # without GPU
Step 12:
sudo make install
Step 13:
Set Environment Variables:
export PYTHONPATH=/usr/local:/usr/local/caffe2:$PYTHONPATH
export LD_LIBRARY_PATH=/usr/local/lib:$LD_LIBRARY_PATH
Step 14:
sudo chown XXX /usr/local/caffe2/python/models
完成安裝後caffe2 會安裝在:/usr/local/caffe2
下載的models放在:/usr/local/caffe2/python/models