diff --git a/pyproject.toml b/pyproject.toml index 3336fc542..259cbf7b2 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -32,7 +32,7 @@ dependencies = [ "python-multipart>=0.0.16", "filelock", "psutil", - "gpuhunt==0.1.17", + "gpuhunt==0.1.18", "argcomplete>=3.5.0", "ignore-python>=0.2.0", "orjson", 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