fix: pytorch version change

mlcommons · Nov 14, 2024 · 67b9f15 · 67b9f15
1 parent 21fb3f9
commit 67b9f15
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -141,15 +141,15 @@ jax_gpu =
 
 # PyTorch CPU
 pytorch_cpu =
-  torch==2.5.0
-  torchvision==0.20.0
+  torch==2.5.1
+  torchvision==0.20.1
 
 # PyTorch GPU
 # Note: omit the cuda suffix and installing from the appropriate
 # wheel will result in using locally installed CUDA.
 pytorch_gpu =
-  torch==2.5.0
-  torchvision==0.20.0
+  torch==2.5.1
+  torchvision==0.20.1
 
 
 ###############################################################################