This module adds the Caffe2 to tp_pipeline. To use this you will need to build and install the Caffe2.

Building Caffe2 Fedora

Dependencies

dnf -y update

dnf -y install \
	glog-devel \
	gtest-devel \
	leveldb-devel \
	lmdb-devel \
	opencv-devel \
	openmpi-devel \
	snappy-devel \
	protobuf-devel \
	protobuf-compiler \
	gflags-devel \
	python3-devel \
	python3-future \
	python3-numpy \
	python3-protobuf \
	python3-hypothesis \
	python3-mypy

CUDA

• Install CUDA

Build

Following instructions on (with some modifications):

• https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=compile
• https://github.com/leonardvandriel/caffe2_cpp_tutorial

Notes:

• We run build rather than install, as we don't want to interfere with the system python.
• Run commands from the directory to which you'll be installing Caffe2; if you build it in one directory and move it afterwards, it may fail at runtime due to missing .so files.

mkdir caffe2
cd caffe2/

mkdir usr
mkdir usr/lib/

git clone https://github.com/pytorch/pytorch.git --recursive 
cd pytorch
python3 setup.py build

cp torch/lib/*.so ../usr/lib/
cp torch/lib/*.a ../usr/lib/
cp -r build/lib/*.a ../usr/lib/
cp -r build/lib/*.so ../usr/lib/
#cp -r torch/lib/include/ ../usr/
cp -r torch/include/ ../usr/

cd ../..

Building Caffe2 iOS

Start by opening XCode and building a test app on your device, this is just to ensure that toolchains are installed.

On a fresh install I needed to reboot after installing XCode.

Then:

brew install automake libtool cmake

mkdir caffe2
cd caffe2

git clone --recursive https://github.com/pytorch/pytorch.git
cd pytorch/
git submodule update --init
./scripts/build_ios.sh