File tree Expand file tree Collapse file tree 1 file changed +5
-0
lines changed
src/dstack/_internal/core/backends/runpod Expand file tree Collapse file tree 1 file changed +5
-0
lines changed Original file line number Diff line number Diff line change 33from typing import Any , Dict , List , Optional
44
55import requests
6+ from gpuhunt .providers .runpod import RunpodProvider
67from requests import Response
78
89from 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"""
You can’t perform that action at this time.
0 commit comments