From d3fc69e4930c34a61a7c67beaa150a50f08ce6d3 Mon Sep 17 00:00:00 2001 From: Zongheng Yang Date: Fri, 8 Mar 2024 16:49:34 -0800 Subject: [PATCH] [RunPod] Fix multi-gpu provisioning. (#3291) --- sky/provision/runpod/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sky/provision/runpod/utils.py b/sky/provision/runpod/utils.py index 00b24aee0a8..9ef35b809fe 100644 --- a/sky/provision/runpod/utils.py +++ b/sky/provision/runpod/utils.py @@ -108,6 +108,7 @@ def launch(name: str, instance_type: str, region: str, disk_size: int) -> str: container_disk_in_gb=disk_size, min_vcpu_count=4 * gpu_quantity, min_memory_in_gb=gpu_specs['memoryInGb'] * gpu_quantity, + gpu_count=gpu_quantity, country_code=region, ports=(f'22/tcp,' f'{constants.SKY_REMOTE_RAY_DASHBOARD_PORT}/http,'