upgrade torch and torchvision

mlcommons · Nov 7, 2023 · 5009ea4 · 5009ea4
1 parent b8afd87
commit 5009ea4
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -131,13 +131,13 @@ jax_gpu =
 
 # PyTorch CPU
 pytorch_cpu =
-  torch==2.0.1
-  torchvision==0.15.2
+  torch==2.1.0
+  torchvision=0.16.0
 
 # PyTorch GPU
 pytorch_gpu =
-  torch==2.0.1+cu118
-  torchvision==0.15.2+cu118
+  torch==2.1.0+cu118
+  torchvision==0.16.0+cu118
 
 # wandb
 wandb =