Praneethcruzon

This can be solved by building the Dockerfile using the Makefile. The argument to the Dockerfile should be as follows: `ARG USE_CUDA=1` `ARG TORCH_ARCH="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"` `ENV TORCH_CUDA_ARCH_LIST` (I manually specified these...