diff --git a/get_torch_cuda_version.py b/get_torch_cuda_version.py index 2b34862..c8a8adf 100644 --- a/get_torch_cuda_version.py +++ b/get_torch_cuda_version.py @@ -10,6 +10,7 @@ support_cuda_versions = { "2.6": [118, 124, 126], "2.7": [118, 126, 128], "2.8": [126, 128, 129], + "2.9": [126, 128, 129], } cuda_version = int(sys.argv[1])