Georg commited on
Commit
c23b196
·
1 Parent(s): 053c7f6

Optimized Docker build to fix OOM errors

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -7,7 +7,7 @@ ENV FOUNDATIONPOSE_MODEL_REPO=gpue/foundationpose-weights
7
  ENV USE_REAL_MODEL=true
8
 
9
  # Ensure NumPy 1.x for CUDA extension compatibility
10
- RUN pip install --no-cache-dir "numpy<2"
11
 
12
  # Set MAX_JOBS=1 BEFORE any CUDA compilation to limit memory usage
13
  ENV MAX_JOBS=1
 
7
  ENV USE_REAL_MODEL=true
8
 
9
  # Ensure NumPy 1.x for CUDA extension compatibility
10
+ RUN pip install --no-cache-dir "numpy<2" transformers
11
 
12
  # Set MAX_JOBS=1 BEFORE any CUDA compilation to limit memory usage
13
  ENV MAX_JOBS=1