Skip to content

Commit 4f9cfee

Browse files
[Runpod] Require CUDA 12.8+ on the host #3299
1 parent a172672 commit 4f9cfee

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/dstack/_internal/core/backends/runpod/api_client.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
from typing import Any, Dict, List, Optional
44

55
import requests
6+
from gpuhunt.providers.runpod import RunpodProvider
67
from requests import Response
78

89
from dstack._internal.core.errors import BackendError, BackendInvalidCredentialsError
@@ -475,6 +476,8 @@ def _generate_pod_deployment_mutation(
475476
)
476477
input_fields.append(f"allowedCudaVersions: [{allowed_cuda_versions_string}]")
477478

479+
input_fields.append(f'minCudaVersion: "{RunpodProvider.MIN_CUDA_VERSION}"')
480+
478481
pod_deploy = "podFindAndDeployOnDemand" if bid_per_gpu is None else "podRentInterruptable"
479482
# Format input fields
480483
input_string = ", ".join(input_fields)
@@ -597,6 +600,8 @@ def _generate_create_cluster_mutation(
597600
ports = ports.replace(" ", "")
598601
input_fields.append(f'ports: "{ports}"')
599602

603+
input_fields.append(f'minCudaVersion: "{RunpodProvider.MIN_CUDA_VERSION}"')
604+
600605
# Format input fields
601606
input_string = ", ".join(input_fields)
602607
return f"""

0 commit comments

Comments
 (0)