feat: add labelling the node with nvidia driver version (#36)

gardenlinux · Jan 23, 2025 · 7ccdb26 · 7ccdb26
1 parent d096c42
commit 7ccdb26
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 2 deletions.
diff --git a/helm/Chart.yaml b/helm/Chart.yaml
@@ -2,4 +2,4 @@ apiVersion: v1
 description: >
   A Helm chart to install NVIDIA GPU Support on Garden Linux-based Gardener clusters
 name: nvidia-installer
-version: 0.0.4
+version: 0.0.5
diff --git a/helm/templates/ds-nvidia-installer.yaml b/helm/templates/ds-nvidia-installer.yaml
@@ -64,7 +64,28 @@ spec:
           mountPath: /opt/nvidia-installer/cache
         - name: module-install-dir-base
           mountPath: {{ required "nvidiaInstaller.hostDriverPath" .Values.nvidiaInstaller.hostDriverPath }}
-      {{- if not .Values.debug }}
+      - image: {{ .Values.nvidiaInstaller.kubectlImage }}
+        name: label-with-nvidia-driver-version
+        securityContext:
+          runAsUser: 65534
+          runAsGroup: 65534
+          runAsNonRoot: true
+          allowPrivilegeEscalation: false
+          capabilities:
+            drop:
+              - "ALL"
+        command:
+          - "sh"
+          - "-c"
+        args:
+          - >
+            echo $NODENAME; kubectl label node $NODENAME ai.sap.com/nvidia-driver-version={{.Values.nvidiaDevicePlugin.nvidiaDriverVersion}} --overwrite
+        env:
+          - name: NODENAME
+            valueFrom:
+              fieldRef:
+                fieldPath: spec.nodeName
+    {{- if not .Values.debug }}
       containers:
       {{- end }}
       - name: fabricmanager

diff --git a/helm/values.yaml b/helm/values.yaml
@@ -43,6 +43,7 @@ nvidiaInstaller:
   # TODO: Edit image tag
   installerTag: latest
   pauseImage: gcr.io/google_containers/pause-amd64:3.0
+  kubectlImage: bitnami/kubectl:latest
   hostDriverPath: /opt/drivers
   installerResources:
     requests: