Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
|
@@ -7,6 +7,9 @@ ENV DEBIAN_FRONTEND=noninteractive
|
|
| 7 |
# Tell PyTorch which GPU architectures to compile for explicitly
|
| 8 |
ENV TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6 8.9 9.0"
|
| 9 |
|
|
|
|
|
|
|
|
|
|
| 10 |
# Set the working directory
|
| 11 |
WORKDIR /app
|
| 12 |
|
|
@@ -31,7 +34,7 @@ RUN pip install --no-cache-dir -r /app/Hunyuan3D-2/requirements.txt
|
|
| 31 |
COPY requirements.txt .
|
| 32 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 33 |
|
| 34 |
-
#
|
| 35 |
RUN pip install --no-cache-dir "numpy<2"
|
| 36 |
|
| 37 |
# Compile the Custom Rasterizer (Required for Textures)
|
|
|
|
| 7 |
# Tell PyTorch which GPU architectures to compile for explicitly
|
| 8 |
ENV TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6 8.9 9.0"
|
| 9 |
|
| 10 |
+
# 🛑 THE FIX: Throttle the compiler to prevent Out Of Memory (OOMKilled) crashes
|
| 11 |
+
ENV MAX_JOBS=2
|
| 12 |
+
|
| 13 |
# Set the working directory
|
| 14 |
WORKDIR /app
|
| 15 |
|
|
|
|
| 34 |
COPY requirements.txt .
|
| 35 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 36 |
|
| 37 |
+
# Explicitly force NumPy to stay below 2.0 right before compiling
|
| 38 |
RUN pip install --no-cache-dir "numpy<2"
|
| 39 |
|
| 40 |
# Compile the Custom Rasterizer (Required for Textures)
|