nfd-nvidia

rwlove · Dec 31, 2024 · 414b217 · 414b217
1 parent 06c31df
commit 414b217
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 3 deletions.
diff --git a/kubernetes/main/apps/kube-system/node-feature-discovery/rules/kustomization.yaml b/kubernetes/main/apps/kube-system/node-feature-discovery/rules/kustomization.yaml
@@ -10,3 +10,4 @@ resources:
   - ./intel-gpu.yaml
   - ./vlan-device.yaml
   - ./barcode-device.yaml
+  - ./nvidia-gpu.yaml
diff --git a/kubernetes/main/apps/kube-system/node-feature-discovery/rules/nvidia-gpu.yaml b/kubernetes/main/apps/kube-system/node-feature-discovery/rules/nvidia-gpu.yaml
@@ -0,0 +1,14 @@
+---
+apiVersion: nfd.k8s-sigs.io/v1alpha1
+kind: NodeFeatureRule
+metadata:
+  name: nvidia-gpu-plugin
+spec:
+  rules:
+    - name: nvidia.gpu
+      labels:
+        nvidia.feature.node.kubernetes.io/gpu: "true"
+      matchFeatures:
+        - feature: pci.device
+          matchExpressions:
+            vendor: { op: In, value: ["10de"] }
diff --git a/kubernetes/main/apps/kube-system/nvidia-device-plugin/app/helmrelease.yaml b/kubernetes/main/apps/kube-system/nvidia-device-plugin/app/helmrelease.yaml
@@ -24,15 +24,14 @@ spec:
       retries: 3
 
   upgrade:
-    cleanupOnFail: true
     crds: CreateReplace
     remediation:
       retries: 3
 
   values:
     runtimeClassName: nvidia
     nodeSelector:
-      nvidia.com/gpu.present: "true"
+      nvidia.feature.node.kubernetes.io/gpu: "true"
     config:
       map:
         default: |-

diff --git a/kubernetes/main/apps/kube-system/nvidia-device-plugin/app/kustomization.yaml b/kubernetes/main/apps/kube-system/nvidia-device-plugin/app/kustomization.yaml
@@ -1,5 +1,4 @@
 ---
----
 # yaml-language-server: $schema=https://json.schemastore.org/kustomization
 apiVersion: kustomize.config.k8s.io/v1beta1
 kind: Kustomization