diff --git a/sky/utils/kubernetes_utils.py b/sky/utils/kubernetes_utils.py index e4a5d27be7a..84f4a8bffff 100644 --- a/sky/utils/kubernetes_utils.py +++ b/sky/utils/kubernetes_utils.py @@ -79,10 +79,10 @@ def get_gke_accelerator_name(accelerator: str) -> str: """Returns the accelerator name for GKE clusters Uses the format - nvidia-tesla-. - A100-80GB and L4 are an exception - they use nvidia-. + A100-80GB, H100-80GB and L4 are an exception. They use nvidia-. """ - if accelerator in ('A100-80GB', 'L4'): - # A100-80GB and L4 have a different name pattern. + if accelerator in ('A100-80GB', 'L4', 'H100-80GB'): + # A100-80GB, L4 and H100-80GB have a different name pattern. return 'nvidia-{}'.format(accelerator.lower()) else: return 'nvidia-tesla-{}'.format(accelerator.lower())