Update torch

fhswf · Nov 7, 2024 · 0750569 · 0750569
1 parent 0415b3a
commit 0750569
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/images/deepml/Dockerfile b/images/deepml/Dockerfile
@@ -67,7 +67,8 @@ RUN pip install \
 
 # Install PyTorch
 RUN pip install \
-    torch==2.4.0+cu124 torchvision==0.19.0+cu124 torchaudio==2.4.0+cu124 --index-url https://download.pytorch.org/whl/cu124 \
+    torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu124
+    #torch==2.4.0+cu124 torchvision==0.19.0+cu124 torchaudio==2.4.0+cu124 --index-url https://download.pytorch.org/whl/cu124 \
     #torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117 \
     && rm -rf /tmp/* $HOME/.cache