Skip to content

Commit

Permalink
[k8s] Fix GPU labeling for EKS (skypilot-org#4146)
Browse files Browse the repository at this point in the history
Fix GPU labelling
  • Loading branch information
romilbhardwaj authored and AlexCuadron committed Oct 24, 2024
1 parent 7e36c4d commit 2918b53
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions sky/utils/kubernetes/k8s_gpu_labeler_job.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit 2918b53

Please sign in to comment.