From 2918b53690a2717d743deda6a18c0909b2edc96b Mon Sep 17 00:00:00 2001
From: Romil Bhardwaj <romil.bhardwaj@berkeley.edu>
Date: Tue, 22 Oct 2024 14:57:48 -0700
Subject: [PATCH] [k8s] Fix GPU labeling for EKS (#4146)

Fix GPU labelling
---
 sky/utils/kubernetes/k8s_gpu_labeler_job.yaml | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/sky/utils/kubernetes/k8s_gpu_labeler_job.yaml b/sky/utils/kubernetes/k8s_gpu_labeler_job.yaml
index f2283bab4c3..78ad13facdc 100644
--- a/sky/utils/kubernetes/k8s_gpu_labeler_job.yaml
+++ b/sky/utils/kubernetes/k8s_gpu_labeler_job.yaml
@@ -14,9 +14,10 @@ spec:
       containers:
       - name: gpu-labeler
         image: us-central1-docker.pkg.dev/skypilot-375900/skypilotk8s/skypilot-gpu:latest # Using this image also serves as a way to "pre-pull" the image onto nodes
-        command:
-          - "python"
-          - "/label_gpus.py"
+        command: ["/bin/bash", "-i", "-c"]
+        args:
+          - |
+            python /label_gpus.py
         env:
         - name: MY_NODE_NAME
           valueFrom: