fix torchvision

mosaicml · Feb 26, 2024 · f103011 · f103011
1 parent 0b1c2de
commit f103011
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/generate_build_matrix.py b/docker/generate_build_matrix.py
@@ -24,7 +24,7 @@
 
 def _get_torchvision_version(pytorch_version: str):
     if pytorch_version == '2.2.1':
-        return '0.17.0'
+        return '0.17.1'
     if pytorch_version == '2.1.2':
         return '0.16.2'
     if pytorch_version == '2.0.1':