Update Dockerfile

Dockerfile

@@ -7,8 +7,13 @@ ENV PYOPENGL_PLATFORM=egl
 ENV CUDA_HOME=/usr/local/cuda
 ENV PATH=${CUDA_HOME}/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
-ENV TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;8.9;9.0"
+
+ENV TORCH_CUDA_ARCH_LIST="8.9"
+ENV CUDAARCHS="89"
 ENV FORCE_CUDA=1
+ENV MAX_JOBS=4
+ENV CMAKE_BUILD_PARALLEL_LEVEL=4
+ENV TORCH_EXTENSIONS_DIR=/workspace/torch_extensions
 
 WORKDIR /workspace
 
@@ -85,9 +90,11 @@ print("CUDA:", torch.version.cuda)
 PY
 
 RUN cd hy3dpaint/custom_rasterizer && \
-    pip install --no-build-isolation -e .
+    MAX_JOBS=4 CUDAARCHS=89 TORCH_CUDA_ARCH_LIST="8.9" \
+    pip install --no-build-isolation --no-cache-dir .
 
 RUN cd hy3dpaint/DifferentiableRenderer && \
+    MAX_JOBS=4 CUDAARCHS=89 TORCH_CUDA_ARCH_LIST="8.9" \
     bash compile_mesh_painter.sh
 
 RUN cat > /workspace/health_server.py <<'PY'