Skip to content

Commit 86a4639

Browse files
Fix Crusoe CPU instances and add H200/B200 support (#3619)
* 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
1 parent 885521d commit 86a4639

2 files changed

Lines changed: 3 additions & 4 deletions

File tree

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ dependencies = [
3232
"python-multipart>=0.0.16",
3333
"filelock",
3434
"psutil",
35-
"gpuhunt==0.1.17",
35+
"gpuhunt==0.1.18",
3636
"argcomplete>=3.5.0",
3737
"ignore-python>=0.2.0",
3838
"orjson",

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

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -91,18 +91,17 @@
9191
IMAGE_SXM_DOCKER = "ubuntu22.04-nvidia-sxm-docker:latest"
9292
IMAGE_PCIE_DOCKER = "ubuntu22.04-nvidia-pcie-docker:latest"
9393
IMAGE_ROCM = "ubuntu-rocm:latest"
94-
IMAGE_BASE = "ubuntu22.04:latest"
9594

9695

9796
def _get_image(instance_name: str, gpu_type: str) -> str:
98-
if not gpu_type:
99-
return IMAGE_BASE
10097
# Check instance name for SXM -- gpu_type from gpuhunt is normalized (e.g. "A100")
10198
# and doesn't contain "SXM", but instance names like "a100-80gb-sxm-ib.8x" do.
10299
if "-sxm" in instance_name.lower():
103100
return IMAGE_SXM_DOCKER
104101
if "MI3" in gpu_type:
105102
return IMAGE_ROCM
103+
# Use PCIe docker image for both PCIe GPUs and CPU-only types.
104+
# Crusoe has no CPU-specific Docker image; the base ubuntu image lacks Docker.
106105
return IMAGE_PCIE_DOCKER
107106

108107

0 commit comments

Comments
 (0)