diff --git a/pkg/config/torch_compatibility_matrix.json b/pkg/config/torch_compatibility_matrix.json index b9db732f5a..4198696fa3 100644 --- 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" ] } -] \ No newline at end of file +]