jayman commited on
Commit
88bb83f
·
verified ·
1 Parent(s): 48fdabb

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -4
Dockerfile CHANGED
@@ -11,8 +11,8 @@ ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
11
  ENV TORCH_CUDA_ARCH_LIST="8.9"
12
  ENV CUDAARCHS="89"
13
  ENV FORCE_CUDA=1
14
- ENV MAX_JOBS=4
15
- ENV CMAKE_BUILD_PARALLEL_LEVEL=4
16
  ENV TORCH_EXTENSIONS_DIR=/workspace/torch_extensions
17
 
18
  WORKDIR /workspace
@@ -90,11 +90,11 @@ print("CUDA:", torch.version.cuda)
90
  PY
91
 
92
  RUN cd hy3dpaint/custom_rasterizer && \
93
- MAX_JOBS=4 CUDAARCHS=89 TORCH_CUDA_ARCH_LIST="8.9" \
94
  pip install --no-build-isolation --no-cache-dir .
95
 
96
  RUN cd hy3dpaint/DifferentiableRenderer && \
97
- MAX_JOBS=4 CUDAARCHS=89 TORCH_CUDA_ARCH_LIST="8.9" \
98
  bash compile_mesh_painter.sh
99
 
100
  RUN cat > /workspace/health_server.py <<'PY'
 
11
  ENV TORCH_CUDA_ARCH_LIST="8.9"
12
  ENV CUDAARCHS="89"
13
  ENV FORCE_CUDA=1
14
+ ENV MAX_JOBS=1
15
+ ENV CMAKE_BUILD_PARALLEL_LEVEL=1
16
  ENV TORCH_EXTENSIONS_DIR=/workspace/torch_extensions
17
 
18
  WORKDIR /workspace
 
90
  PY
91
 
92
  RUN cd hy3dpaint/custom_rasterizer && \
93
+ MAX_JOBS=1 CUDAARCHS=89 TORCH_CUDA_ARCH_LIST="8.9" \
94
  pip install --no-build-isolation --no-cache-dir .
95
 
96
  RUN cd hy3dpaint/DifferentiableRenderer && \
97
+ MAX_JOBS=1 CUDAARCHS=89 TORCH_CUDA_ARCH_LIST="8.9" \
98
  bash compile_mesh_painter.sh
99
 
100
  RUN cat > /workspace/health_server.py <<'PY'