Spaces:

fffiloni
/

KEEP-docker

Runtime error

fffiloni commited on Jan 15

Commit

0db000b

verified ·

1 Parent(s): 9d8aa61

Update Dockerfile

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -50,8 +50,9 @@ ENV HOME=/home/user \
 # Set working directory
 WORKDIR $HOME/app
-# Upgrade pip and install PyTorch first
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir torch==1.7.1 torchvision --extra-index-url https://download.pytorch.org/whl/cu113 && \
     python -c "import torch; print('PyTorch version:', torch.__version__)"

 # Set working directory
 WORKDIR $HOME/app
+# Upgrade pip and install numpy<2 first, then PyTorch
 RUN pip install --no-cache-dir --upgrade pip && \
+    pip install --no-cache-dir "numpy<2" && \
     pip install --no-cache-dir torch==1.7.1 torchvision --extra-index-url https://download.pytorch.org/whl/cu113 && \
     python -c "import torch; print('PyTorch version:', torch.__version__)"