Skip to content

Commit 769de74

Browse files
committed
test for pytorch
1 parent ce25cd3 commit 769de74

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

Framework/pytorch/Dockerfile

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

0 commit comments

Comments
 (0)