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

Author: peterschmidt85

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"
 

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