diff --git a/dialogs/packages_installer/packages_installer_dialog.py b/dialogs/packages_installer/packages_installer_dialog.py index d2d62be0d7bcc1a0d25c485b593ac05d79f96a36..a061e63965ebb1dad585a59c9a2cb7331b2cb5c6 100644 --- a/dialogs/packages_installer/packages_installer_dialog.py +++ b/dialogs/packages_installer/packages_installer_dialog.py @@ -394,6 +394,7 @@ def get_pytorch_version(cuda_version): cuda_to_pytorch = { "11.8": " --index-url https://download.pytorch.org/whl/cu118", "12.1": "", + "12.2": "", "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",