Support torch 2.5.1 (#2070)

replicate · Nov 29, 2024 · 51b6f44 · 51b6f44
1 parent ed0fb02
commit 51b6f44
Showing 1 changed file with 63 additions and 2 deletions.
diff --git a/pkg/config/torch_compatibility_matrix.json b/pkg/config/torch_compatibility_matrix.json
@@ -1,7 +1,68 @@
 [
+  {
+    "Torch": "2.5.1+cpu",
+    "Torchvision": "0.20.1",
+    "Torchaudio": "2.5.1",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cpu",
+    "CUDA": null,
+    "Pythons": [
+      "3.10",
+      "3.11",
+      "3.12",
+      "3.13",
+      "3.9"
+    ]
+  },
+  {
+    "Torch": "2.5.1+cu118",
+    "Torchvision": "0.20.1",
+    "Torchaudio": "2.5.1",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cu118",
+    "CUDA": "11.8",
+    "Pythons": [
+      "3.10",
+      "3.11",
+      "3.12",
+      "3.13",
+      "3.9"
+    ]
+  },
+  {
+    "Torch": "2.5.1+cu121",
+    "Torchvision": "0.20.1",
+    "Torchaudio": "2.5.1",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cu121",
+    "CUDA": "12.1",
+    "Pythons": [
+      "3.10",
+      "3.11",
+      "3.12",
+      "3.13",
+      "3.9",
+      "3.10"
+    ]
+  },
+  {
+    "Torch": "2.5.1+cu124",
+    "Torchvision": "0.20.1",
+    "Torchaudio": "2.5.1",
+    "FindLinks": "",
+    "ExtraIndexURL": "https://download.pytorch.org/whl/cu124",
+    "CUDA": "12.4",
+    "Pythons": [
+      "3.10",
+      "3.11",
+      "3.12",
+      "3.13",
+      "3.9"
+    ]
+  },
   {
     "Torch": "2.5.0+cpu",
-    "Torchvision": "0.20.0",
+    "Torchvision": "0.20.1",
     "Torchaudio": "2.5.0",
     "FindLinks": "",
     "ExtraIndexURL": "https://download.pytorch.org/whl/cpu",
@@ -1590,4 +1651,4 @@
       "3.12"
     ]
   }
-]
+]