From f82c2a178b2f54c4781021bed2485866b3980475 Mon Sep 17 00:00:00 2001 From: Andrew Date: Sun, 5 May 2024 23:38:56 -0700 Subject: [PATCH] add P4000 --- sky/provision/kubernetes/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sky/provision/kubernetes/utils.py b/sky/provision/kubernetes/utils.py index 7ab8265ec54..e435a8326e3 100644 --- a/sky/provision/kubernetes/utils.py +++ b/sky/provision/kubernetes/utils.py @@ -172,7 +172,7 @@ def get_accelerator_from_label_value(cls, value: str) -> str: """ canonical_gpu_names = [ 'A100-80GB', 'A100', 'A10G', 'H100', 'K80', 'M60', 'T4g', 'T4', - 'V100', 'A10', 'P100', 'P40', 'P4', 'L4' + 'V100', 'A10', 'P4000', 'P100', 'P40', 'P4', 'L4' ] for canonical_name in canonical_gpu_names: # A100-80G accelerator is A100-SXM-80GB or A100-PCIE-80GB