Android를 위한 Caffe2 빌드

Ubuntu기준으로 설명.

일단 caffe2를 빌드하기 위해서 필요한 package들을 설치.

sudo apt-get update sudo apt-get install -y --no-install-recommends build-essential git libgoogle-glog-dev libgtest-dev libiomp-dev libleveldb-dev liblmdb-dev libopencv-dev libopenmpi-dev libsnappy-dev libprotobuf-dev openmpi-bin openmpi-doc protobuf-compiler python-dev python-pip

pip install --user future numpy protobuf typing hypothesis

Caffe2 Source를 다운로드하고 빌드함.

git clone https://github.com/pytorch/pytorch.git && cd pytorch
git submodule update --init --recursive
python setup.py install

cd ~ && python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

Android SDK 및 NDK를 설치.
$ export ANDROID_SDK=/home/user/Android/Sdk
$ export ANDROID_NDK=/home/user/Android/Sdk/ndk-bundle
$ export PATH=$PATH:$ANDROID_SDK/platform-tools
$ export PATH=$PATH:$ANDROID_SDK/tools
$ export PATH=$PATH:$ANDROID_ND

cd script
vi ./build_android.sh
현재 armeabi-v7a으로 설정되어 있기때문에 arm64-v8a로 변경 및 사용중인 SDK version으로 변경.
./build_android.sh

https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=compile

pytorch/pytorch