From 1bbe3d98aa0cba74cab22104e552895270284135 Mon Sep 17 00:00:00 2001 From: peterschmidt85 Date: Thu, 26 Feb 2026 21:24:00 +0100 Subject: [PATCH 1/2] Fix Crusoe CPU instances and add H200/B200 support - Use docker-enabled VM image for CPU types (base image lacks Docker) - Update gpuhunt with H200, B200, and CPU instance support Made-with: Cursor --- pyproject.toml | 6 +++++- src/dstack/_internal/core/backends/crusoe/compute.py | 5 ++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 3336fc542..810b8a540 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,7 +32,8 @@ dependencies = [ "python-multipart>=0.0.16", "filelock", "psutil", - "gpuhunt==0.1.17", + # TODO: Pin to gpuhunt release once crusoe-add-new-gpu-types branch is merged. + "gpuhunt @ git+https://github.com/dstackai/gpuhunt.git@crusoe-add-new-gpu-types", "argcomplete>=3.5.0", "ignore-python>=0.2.0", "orjson", @@ -54,6 +55,9 @@ build-backend = "hatchling.build" [project.scripts] dstack = "dstack._internal.cli.main:main" +[tool.hatch.metadata] +allow-direct-references = true + [tool.hatch.version] path = "src/dstack/version.py" diff --git a/src/dstack/_internal/core/backends/crusoe/compute.py b/src/dstack/_internal/core/backends/crusoe/compute.py index 10ede9677..7de7d0967 100644 --- a/src/dstack/_internal/core/backends/crusoe/compute.py +++ b/src/dstack/_internal/core/backends/crusoe/compute.py @@ -91,18 +91,17 @@ IMAGE_SXM_DOCKER = "ubuntu22.04-nvidia-sxm-docker:latest" IMAGE_PCIE_DOCKER = "ubuntu22.04-nvidia-pcie-docker:latest" IMAGE_ROCM = "ubuntu-rocm:latest" -IMAGE_BASE = "ubuntu22.04:latest" def _get_image(instance_name: str, gpu_type: str) -> str: - if not gpu_type: - return IMAGE_BASE # Check instance name for SXM -- gpu_type from gpuhunt is normalized (e.g. "A100") # and doesn't contain "SXM", but instance names like "a100-80gb-sxm-ib.8x" do. if "-sxm" in instance_name.lower(): return IMAGE_SXM_DOCKER if "MI3" in gpu_type: return IMAGE_ROCM + # Use PCIe docker image for both PCIe GPUs and CPU-only types. + # Crusoe has no CPU-specific Docker image; the base ubuntu image lacks Docker. return IMAGE_PCIE_DOCKER From 8b5c2168b72540050e9e4ebfde29f50ce5df54f1 Mon Sep 17 00:00:00 2001 From: Andrey Cheptsov Date: Tue, 3 Mar 2026 11:39:20 +0100 Subject: [PATCH 2/2] Update gpuhunt dependency to version 0.1.18 --- pyproject.toml | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 810b8a540..259cbf7b2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,8 +32,7 @@ dependencies = [ "python-multipart>=0.0.16", "filelock", "psutil", - # TODO: Pin to gpuhunt release once crusoe-add-new-gpu-types branch is merged. - "gpuhunt @ git+https://github.com/dstackai/gpuhunt.git@crusoe-add-new-gpu-types", + "gpuhunt==0.1.18", "argcomplete>=3.5.0", "ignore-python>=0.2.0", "orjson", @@ -55,9 +54,6 @@ build-backend = "hatchling.build" [project.scripts] dstack = "dstack._internal.cli.main:main" -[tool.hatch.metadata] -allow-direct-references = true - [tool.hatch.version] path = "src/dstack/version.py"