diff --git a/dialogs/packages_installer/packages_installer_dialog.py b/dialogs/packages_installer/packages_installer_dialog.py index f6b471969fc262fa24b9ca531dc9173813b5f0c7..7a0cde4ea94fa2e59ef68821ec3534ac3be6195c 100644 --- a/dialogs/packages_installer/packages_installer_dialog.py +++ b/dialogs/packages_installer/packages_installer_dialog.py @@ -321,6 +321,7 @@ def get_pytorch_version(cuda_version): "11.8": " --index-url https://download.pytorch.org/whl/cu118", "12.1": "", "12.4": " --index-url https://download.pytorch.org/whl/cu124", + "12.5": " --index-url https://download.pytorch.org/whl/cu124", "12.6": " --index-url https://download.pytorch.org/whl/cu124", } return cuda_to_pytorch.get(cuda_version, None)