From 1946bd49e869b02ed956dcb6bd5740d44a18a1fc Mon Sep 17 00:00:00 2001
From: Dmitry Meyer <me@undef.im>
Date: Thu, 23 Apr 2026 09:03:52 +0000
Subject: [PATCH] Bump `gpuhunt`, add support for all GeForce RTX 2..5 series

---
 pyproject.toml                        | 2 +-
 src/dstack/_internal/utils/gpu.py     | 1 +
 src/tests/_internal/utils/test_gpu.py | 2 ++
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index aa92ed6185..420b07e80d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -32,7 +32,7 @@ dependencies = [
     "python-multipart>=0.0.16",
     "filelock",
     "psutil",
-    "gpuhunt==0.1.20",
+    "gpuhunt==0.1.21",
     "argcomplete>=3.5.0",
     "ignore-python>=0.2.0",
     "orjson",
diff --git a/src/dstack/_internal/utils/gpu.py b/src/dstack/_internal/utils/gpu.py
index 94a12c6df7..d0fc94e50c 100644
--- a/src/dstack/_internal/utils/gpu.py
+++ b/src/dstack/_internal/utils/gpu.py
@@ -20,6 +20,7 @@ def convert_nvidia_gpu_name(name: str) -> str:
         return name.replace(" ", "")
 
     name = name.replace(" Ti", "Ti")
+    name = re.sub(r"(?i) ?SUPER", "SUPER", name)
     name = name.replace(" NVL", "NVL")
     name = name.replace(" Ada Generation", "Ada")
     name = name.replace("RTX ", "RTX")
diff --git a/src/tests/_internal/utils/test_gpu.py b/src/tests/_internal/utils/test_gpu.py
index c5a58a4ae2..b649b8d61a 100644
--- a/src/tests/_internal/utils/test_gpu.py
+++ b/src/tests/_internal/utils/test_gpu.py
@@ -22,6 +22,8 @@ class TestConvertGpuName:
             ("NVIDIA H100 NVL", "H100NVL"),
             ("NVIDIA H100 80GB HBM3", "H100"),
             ("Tesla T4", "T4"),
+            ("NVIDIA GeForce RTX 2070 Super", "RTX2070SUPER"),
+            ("NVIDIA GeForce RTX 4070 Ti SUPER", "RTX4070TiSUPER"),
         ],
     )
     def test_convert_nvidia_gpu_name(self, test_input, expected):