For me, it was not very trivial to run caffe2 on Android. But, https://github.com/caffe2/AICamera 117 was very helpful.
I’m not sure if this is still valid for PyTorch integration with caffe2 (but it is 1.0, so you seem to use caffe2 separated from PyTorch).