Bump gpuhunt, add support for all GeForce RTX 2..5 series (#3818)

Author: un-def

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",

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")

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):