From d7847260ec0774c4be0cde948be638182c0a8d8b Mon Sep 17 00:00:00 2001
From: Junya Morioka <mjun@mjunya.com>
Date: Sun, 10 Aug 2025 16:56:59 +0900
Subject: [PATCH] update pytorch

---
 get_torch_cuda_version.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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