Refactor Dockerfile to include a timeout for pip3 installation of PyT…

…orch
labring-actions · Dec 10, 2024 · 74bf99c · 74bf99c
1 parent 180f89d
commit 74bf99c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Framework/pytorch/Dockerfile b/Framework/pytorch/Dockerfile
@@ -12,7 +12,7 @@ RUN apt-get update && \
     . /root/.bashrc && \
     conda create -y -n hello python=3.12 && \
     conda activate hello && \
-    pip3 install torch torchvision torchaudio && \
+    pip3 install --timeout 10000 torch torchvision torchaudio && \
     chown -R devbox:devbox /home/devbox/project && \
     chmod -R u+rw /home/devbox/project && \
     chmod -R +x /home/devbox/project/entrypoint.sh && \