Spaces:
Build error
Build error
Update Dockerfile
Browse files- 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=
|
| 15 |
-
ENV CMAKE_BUILD_PARALLEL_LEVEL=
|
| 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=
|
| 94 |
pip install --no-build-isolation --no-cache-dir .
|
| 95 |
|
| 96 |
RUN cd hy3dpaint/DifferentiableRenderer && \
|
| 97 |
-
MAX_JOBS=
|
| 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'
|