[k8s] Kubernetes H100-80GB support (#2840)

* H100 support * comments
skypilot-org · Dec 5, 2023 · d263122 · d263122
1 parent 04553ad
commit d263122
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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-<accelerator>.
- A100-80GB and L4 are an exception - they use nvidia-<accelerator>.
+ A100-80GB, H100-80GB and L4 are an exception. They use nvidia-<accelerator>.
  """
- 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())