From b888379e8875fe3ffa65a97623dadaf9beaf174c Mon Sep 17 00:00:00 2001 From: vishal Date: Fri, 5 Jun 2020 20:51:52 -0400 Subject: [PATCH] Update generate_eks.py --- manager/generate_eks.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/manager/generate_eks.py b/manager/generate_eks.py index 46e919045c..1cee861d3e 100644 --- a/manager/generate_eks.py +++ b/manager/generate_eks.py @@ -102,8 +102,12 @@ def apply_gpu_settings(nodegroup): "tags": { "k8s.io/cluster-autoscaler/node-template/label/nvidia.com/gpu": "true", "k8s.io/cluster-autoscaler/node-template/taint/dedicated": "nvidia.com/gpu=true", + "k8s.io/cluster-autoscaler/node-template/label/k8s.amazonaws.com/accelerator": "true", # accepted values are GPU type such as nvidia-tesla-k80 but using "true" as a placeholder for now because the value doesn't matter for AWS cluster autoscaler + }, + "labels": { + "nvidia.com/gpu": "true", + "k8s.amazonaws.com/accelerator": "true", # accepted values are GPU type such as nvidia-tesla-k80 but using "true" as a placeholder for now because the value doesn't matter for AWS cluster autoscaler }, - "labels": {"nvidia.com/gpu": "true"}, "taints": {"nvidia.com/gpu": "true:NoSchedule"}, }