Skip to content

Commit 1c07f64

Browse files
[Runpod] Require CUDA 12.8+ on the host (#3304)
* [Runpod] Require CUDA 12.8+ on the host #3299 * [Runpod] Require CUDA 12.8+ on the host #3299 Updated `gpuhunt` to `0.1.13`
1 parent d7ea72c commit 1c07f64

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ dependencies = [
3232
"python-multipart>=0.0.16",
3333
"filelock",
3434
"psutil",
35-
"gpuhunt==0.1.12",
35+
"gpuhunt==0.1.13",
3636
"argcomplete>=3.5.0",
3737
"ignore-python>=0.2.0",
3838
"orjson",

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)