sudo make

On building PyTorch libs in Docker

Recently I've been building containerized apps written in Caffe2/PyTorch. One of them had a dependency on a third-party API with some custom PyTorch modules built via torch.utils.ffi. This is a three-step process:

  1. nvcc compiles the CUDA code and builds a shared object.
  2. PyTorch utils create an FFI
Continue Reading