vish85521 commited on
Commit
d5e87bc
·
verified ·
1 Parent(s): c67171e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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
- # 🛑 THE FIX: Explicitly force NumPy to stay below 2.0 right before compiling
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)