Skip to content
Snippets Groups Projects
Commit 2bda17a2 authored by paul.tresson_ird.fr's avatar paul.tresson_ird.fr
Browse files

test for cuda 12.2

parent 7c8be1e9
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
dialogs/packages_installer/packages_installer_dialog.py
+ 1
− 0
View file @ 2bda17a2
......@@ -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",
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment