[Bug] Several issues with vastai provider #142 #2566 (#2567)

peterschmidt85 · web-flow · commit c5f207dafb77 · 2025-04-30T10:54:26.000+02:00
* [Bug] Several issues with `vastai` provider #142 #2566 (passing `cuda_max_good` and `compute_cap`) * [Bug] Several issues with `vastai` provider #142 #2566 (migrated to `gpuhunt==0.1.5`)
diff --git a/pyproject.toml b/pyproject.toml
@@ -32,7 +32,7 @@ dependencies = [
     "python-multipart>=0.0.16",
     "filelock",
     "psutil",
-    "gpuhunt>=0.1.3,<0.2.0",
+    "gpuhunt==0.1.5",
     "argcomplete>=3.5.0",
 ]
 
diff --git a/src/dstack/_internal/core/backends/vastai/compute.py b/src/dstack/_internal/core/backends/vastai/compute.py
@@ -43,7 +43,8 @@ def __init__(self, config: VastAIConfig):
                     "reliability2": {"gte": 0.9},
                     "inet_down": {"gt": 128},
                     "verified": {"eq": True},
-                    "cuda_max_good": {"gte": 11.8},
+                    "cuda_max_good": {"gte": 12.1},
+                    "compute_cap": {"gte": 600},
                 }
             )
         )

Original file line number	Diff line number	Diff line change
`@@ -32,7 +32,7 @@ dependencies = [`
`32`	`32`	`"python-multipart>=0.0.16",`
`33`	`33`	`"filelock",`
`34`	`34`	`"psutil",`
`35`		`- "gpuhunt>=0.1.3,<0.2.0",`
	`35`	`+ "gpuhunt==0.1.5",`
`36`	`36`	`"argcomplete>=3.5.0",`
`37`	`37`	`]`
`38`	`38`
Original file line number	Diff line number	Diff line change
`@@ -43,7 +43,8 @@ def __init__(self, config: VastAIConfig):`
`43`	`43`	`"reliability2": {"gte": 0.9},`
`44`	`44`	`"inet_down": {"gt": 128},`
`45`	`45`	`"verified": {"eq": True},`
`46`		`- "cuda_max_good": {"gte": 11.8},`
	`46`	`+ "cuda_max_good": {"gte": 12.1},`
	`47`	`+ "compute_cap": {"gte": 600},`
`47`	`48`	`}`
`48`	`49`	`)`
`49`	`50`	`)`