Skip to content

Commit 27a42b7

Browse files
authored
Use Cuda 12.0 image for DataCrunch A6000 (#3105)
1 parent e9d863c commit 27a42b7

File tree

1 file changed

+3
-4
lines changed
  • src/dstack/_internal/core/backends/datacrunch

1 file changed

+3
-4
lines changed

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

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -185,10 +185,9 @@ def update_provisioning_data(
185185

186186
def _get_vm_image_id(instance_offer: InstanceOfferWithAvailability) -> str:
187187
# https://api.datacrunch.io/v1/images
188-
if (
189-
len(instance_offer.instance.resources.gpus) > 0
190-
and instance_offer.instance.resources.gpus[0].name == "V100"
191-
):
188+
if len(instance_offer.instance.resources.gpus) > 0 and instance_offer.instance.resources.gpus[
189+
0
190+
].name in ["V100", "A6000"]:
192191
# Ubuntu 22.04 + CUDA 12.0 + Docker
193192
return "2088da25-bb0d-41cc-a191-dccae45d96fd"
194193
# Ubuntu 24.04 + CUDA 12.8 Open + Docker

0 commit comments

Comments
 (0)