Merge pull request #221 from johnbelamaric/compute-class-toleration

Add toleration for compute class taint
NVIDIA · Feb 12, 2025 · 8945a29 · 8945a29
2 parents 1056ce0 + 35ba0f4
commit 8945a29
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/demo/clusters/gke/install-dra-driver.sh b/demo/clusters/gke/install-dra-driver.sh
@@ -39,4 +39,7 @@ helm upgrade -i --create-namespace --namespace nvidia nvidia-dra-driver ${PROJEC
   --set nvidiaDriverRoot="/opt/nvidia" \
   --set kubeletPlugin.tolerations[0].key=nvidia.com/gpu \
   --set kubeletPlugin.tolerations[0].operator=Exists \
-  --set kubeletPlugin.tolerations[0].effect=NoSchedule
+  --set kubeletPlugin.tolerations[0].effect=NoSchedule \
+  --set kubeletPlugin.tolerations[1].key=cloud.google.com/compute-class \
+  --set kubeletPlugin.tolerations[1].operator=Exists \
+  --set kubeletPlugin.tolerations[1].effect=NoSchedule