BrainCaffe2

Info on building Caffe 2 from source

Install Caffe2 on Ubuntu (16.04 with GPUs)

Basically, follow official instruction here (with some modifications).

System libraries

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 \
      libgflags-dev \
      cmake

Python packages

sudo pip3 install \
      future \
      numpy \
      protobuf \
      typing \
      hypothesis \
      pyyaml \
      pydot

Build from source

Under ~/src, run the following:

git clone https://github.com/pytorch/pytorch.git && cd pytorch
git checkout db5d313
git submodule update --init --recursive
CC=/usr/bin/gcc-5 CXX=/usr/bin/g++-5 python3 setup.py install

test

Check whether build was successful or not

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

Test a module.

python3 caffe2/python/operator_test/activation_ops_test.py

Detectron installation

pip3 install -r requirements.txt  --user

Install Caffe2 on OSX (10.14 without GPUs)

Install Macports

Install systempackages using Macports

sudo port install cmake gflags

Set up python packages using conda

For python 2.7

conda create -n caffe

conda activate caffe

conda install future \
      numpy \
      protobuf \
      typing \
      hypothesis \
      pyyaml \
      pydot

For Python 3.6

conda create -n caffe36

conda activate caffe36

conda install future \
      numpy \
      protobuf \
      typing \
      hypothesis \
      pyyaml \
      pydot

Get source

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

Build and install

export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install

https://github.com/pytorch/pytorch#from-source

Set up path before launch python

export DYLD_LIBRARY_PATH=`python -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())"`/caffe2/python

After using it, you can unset DYLD_LIBRARY_PATH by

unset DYLD_LIBRARY_PATH

Alternatively, use Docker from Nvidia

ys7yoo / BrainCaffe2

readme