Skip to content

Commit 74bf99c

Browse files
committed
Refactor Dockerfile to include a timeout for pip3 installation of PyTorch
1 parent 180f89d commit 74bf99c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Framework/pytorch/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ RUN apt-get update && \
1212
. /root/.bashrc && \
1313
conda create -y -n hello python=3.12 && \
1414
conda activate hello && \
15-
pip3 install torch torchvision torchaudio && \
15+
pip3 install --timeout 10000 torch torchvision torchaudio && \
1616
chown -R devbox:devbox /home/devbox/project && \
1717
chmod -R u+rw /home/devbox/project && \
1818
chmod -R +x /home/devbox/project/entrypoint.sh && \

0 commit comments

Comments
 (0)