Skip to content

Commit b78851b

Browse files
authored
Adjust kubernetes gpu matching for RTX5090 (#3440)
1 parent 3e931d9 commit b78851b

File tree

1 file changed

+1
-0
lines changed
  • src/dstack/_internal/core/backends/kubernetes

1 file changed

+1
-0
lines changed

src/dstack/_internal/core/backends/kubernetes/compute.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -574,6 +574,7 @@ def _get_nvidia_gpu_from_node_labels(labels: dict[str, str]) -> Optional[Gpu]:
574574
gpu_product = labels.get(NVIDIA_GPU_PRODUCT_LABEL)
575575
if gpu_product is None:
576576
return None
577+
gpu_product = gpu_product.replace("RTX-", "RTX")
577578
for gpu_name in NVIDIA_GPU_NAMES:
578579
if gpu_name.lower() in gpu_product.lower().split("-"):
579580
break

0 commit comments

Comments
 (0)