From ed1fdf6d52bd10aff4321ed09d5a3dfdc812cc06 Mon Sep 17 00:00:00 2001
From: htulet <hadrien.tulet@cirad.fr>
Date: Fri, 8 Nov 2024 11:46:46 +0100
Subject: [PATCH] gpu 12.5
---
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 f6b4719..7a0cde4 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)
--
GitLab