From 9e3c98a7bcd097307ed24bf1832ae941246c87b1 Mon Sep 17 00:00:00 2001 From: Jvst Me Date: Mon, 15 Jun 2026 22:31:43 +0200 Subject: [PATCH] [Nebius]: Update OS image and add new platforms - Update the OS image to `ubuntu24.04-cuda13.0` on all platforms - Add new platforms: - `gpu-rtx6000` - NVIDIA RTX PRO 6000 - `gpu-b200-sxm-a` - NVIDIA B200 with Intel Xeon Platinum 8570 (in addition to the already supported `gpu-b200-sxm` - NVIDIA B200 with Intel Xeon Platinum 8580) --- src/dstack/_internal/core/backends/nebius/compute.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/dstack/_internal/core/backends/nebius/compute.py b/src/dstack/_internal/core/backends/nebius/compute.py index 9d54513390..a90b4a4c1f 100644 --- a/src/dstack/_internal/core/backends/nebius/compute.py +++ b/src/dstack/_internal/core/backends/nebius/compute.py @@ -58,7 +58,7 @@ logger = get_logger(__name__) CONFIGURABLE_DISK_SIZE = Range[Memory]( - min=Memory.parse("40GB"), # min for the ubuntu22.04-cuda12 image + min=Memory.parse("40GB"), # min for the ubuntu24.04-cuda13.0 image max=Memory.parse("8192GB"), # max for the NETWORK_SSD disk type ) WAIT_FOR_DISK_TIMEOUT = 20 @@ -80,8 +80,10 @@ "gpu-h100-sxm", "gpu-h200-sxm", "gpu-b200-sxm", + "gpu-b200-sxm-a", "gpu-l40s-a", "gpu-l40s-d", + "gpu-rtx6000", "cpu-d3", "cpu-e2", ] @@ -167,15 +169,12 @@ def create_instance( resource_tags=instance_config.tags, ) labels = resources.filter_invalid_labels(labels) - gpus = instance_offer.instance.resources.gpus create_disk_op = resources.create_disk( sdk=self._sdk, name=instance_name, project_id=self._region_to_project_id[instance_offer.region], size_mib=instance_offer.instance.resources.disk.size_mib, - image_family="ubuntu24.04-cuda12" - if gpus and gpus[0].name == "B200" - else "ubuntu22.04-cuda12", + image_family="ubuntu24.04-cuda13.0", labels=labels, ) create_instance_op = None