./build.sh lattice_img fails

I'm having some difficultly building the docker image. The final few lines I see are:
```/opt/pytorch/c10/core/MemoryFormat.h:34:47: note: declared here
C10_DEPRECATED inline MemoryFormat get_contiguous_memory_format() {
^~~~~~~~~~~~~~~~~~~~~
[1585/2460] Building NVCC (Device) object caffe2/CMakeFiles/torch.dir/__/aten/src/ATen/native/cuda/torch_generated_BinaryCompareKernel.cu.o
[1586/2460] Building NVCC (Device) object caffe2/CMakeFiles/torch.dir/__/aten/src/ATen/native/cuda/torch_generated_SpectralOps.cu.o
[1587/2460] Building NVCC (Device) object caffe2/CMakeFiles/torch.dir/__/aten/src/ATen/native/cuda/torch_generated_PointwiseOpsKernel.cu.o
The command '/bin/sh -c MAX_JOBS=8 USE_NINJA=1 BUILD_TEST=0 BUILD_JNI=0 python3 setup.py install' returned a non-zero code: 137
```

There are many warnings, but I didn't see any errors.