Skip to content

Commit

Permalink
[RunPod] Fix multi-gpu provisioning. (#3291)
Browse files Browse the repository at this point in the history
  • Loading branch information
concretevitamin authored Mar 9, 2024
1 parent 341ebb6 commit d3fc69e
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions sky/provision/runpod/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,'
Expand Down

0 comments on commit d3fc69e

Please sign in to comment.