From 3948daa7f90e4ed3b1bfba48d46892e82b99830e Mon Sep 17 00:00:00 2001 From: Victor Skvortsov Date: Tue, 16 Sep 2025 12:22:18 +0500 Subject: [PATCH] Use Cuda 12.0 image for DataCrunch A6000 --- src/dstack/_internal/core/backends/datacrunch/compute.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/dstack/_internal/core/backends/datacrunch/compute.py b/src/dstack/_internal/core/backends/datacrunch/compute.py index afc4bf8511..bec8e2b841 100644 --- a/src/dstack/_internal/core/backends/datacrunch/compute.py +++ b/src/dstack/_internal/core/backends/datacrunch/compute.py @@ -185,10 +185,9 @@ def update_provisioning_data( def _get_vm_image_id(instance_offer: InstanceOfferWithAvailability) -> str: # https://api.datacrunch.io/v1/images - if ( - len(instance_offer.instance.resources.gpus) > 0 - and instance_offer.instance.resources.gpus[0].name == "V100" - ): + if len(instance_offer.instance.resources.gpus) > 0 and instance_offer.instance.resources.gpus[ + 0 + ].name in ["V100", "A6000"]: # Ubuntu 22.04 + CUDA 12.0 + Docker return "2088da25-bb0d-41cc-a191-dccae45d96fd" # Ubuntu 24.04 + CUDA 12.8 Open + Docker