Add 2.5.1 PyTorch & Torchvision (#4128)

upgrade torch version
openvinotoolkit · Nov 25, 2024 · 299711e · 299711e
1 parent a92db5d
commit 299711e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/otx/cli/utils/installation.py b/src/otx/cli/utils/installation.py
@@ -32,6 +32,7 @@
     "2.3.0": {"torchvision": "0.18.0", "cuda": ("11.8", "12.1")},
     "2.3.1": {"torchvision": "0.18.1", "cuda": ("11.8", "12.1")},
     "2.4.0": {"torchvision": "0.19.0", "cuda": ("11.8", "12.1", "12.4")},
+    "2.5.1": {"torchvision": "0.20.1", "cuda": ("11.8", "12.1", "12.4")},
 }
 
 MM_REQUIREMENTS = [