/caffe-android-lib-distro

Forked from https://github.com/sh1r0/caffe-android-lib.git and adding distributed training support.

Primary LanguageShellOtherNOASSERTION

Caffe-Android-Lib

Goal

Porting caffe to android platform

Support

  • Up-to-date caffe (d91572d)
  • CPU only
  • Without support for some IO libs (leveldb and hdf5)

Build

Tested with Android NDK r11c and cmake 3.3.2 on Ubuntu 14.04

git clone --recursive https://github.com/huzi96/caffe-android-lib-distro.git
cd caffe-android-lib
export ANDROID_ABI="arm64-v8a" # Optional, see the note below
./build.sh <path/to/ndk>

NOTE: OpenBLAS

OpenBLAS is the only supported BLAS choice now, and the supported ABIs are the following:

  • armeabi
  • armeabi-v7a-hard-softfp with NEON
  • arm64-v8a (default)
  • x86
  • x86_64

Issues

Any comments, issues or PRs are welcomed. Thanks.

TODO

  • Code cleaning and minor bug fix.]

Optional

.envrc files are for direnv

direnv is an environment variable manager for your shell. It knows how to hook into bash, zsh and fish shell to load or unload environment variables depending on your current directory. This allows to have project-specific environment variables and not clutter the "~/.profile" file.