From 2bda17a260fd0b8b1ed4e088f8985f5ff257cd90 Mon Sep 17 00:00:00 2001 From: ptresson <paul.tresson@ird.fr> Date: Fri, 10 Jan 2025 09:17:30 +0100 Subject: [PATCH] test for cuda 12.2 --- dialogs/packages_installer/packages_installer_dialog.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dialogs/packages_installer/packages_installer_dialog.py b/dialogs/packages_installer/packages_installer_dialog.py index d2d62be..a061e63 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", -- GitLab