Update Dockerfile

Dockerfile

@@ -4,9 +4,10 @@ ARG DEBIAN_FRONTEND=noninteractive
 
 ENV PYTHONUNBUFFERED=1
 
-ENV TORCH_CUDA_ARCH_LIST="6.0 6.1 7.0 7.5 8.0 8.6"
-ENV TCNN_CUDA_ARCHITECTURES=86;80;75;70;61;60
+ENV TORCH_CUDA_ARCH_LIST="8.6"
+ENV TCNN_CUDA_ARCHITECTURES=86
 ENV FORCE_CUDA=1
+ENV MAX_JOBS=1
 
 ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:${PATH}