From b1da9db48bf18cd9d1d09d0b08b9f2e791eae59d Mon Sep 17 00:00:00 2001
From: Tyler Perkins <tyler@clortox.com>
Date: Sat, 30 Mar 2024 16:51:25 -0400
Subject: [PATCH] Add nvidia-device-plugin

---
 base/nvidia/nvidia-device-plugin.yaml | 30 +++++++++++++++++++++++++++
 base/{ => nvidia}/nvidia-runtime.yaml |  0
 2 files changed, 30 insertions(+)
 create mode 100644 base/nvidia/nvidia-device-plugin.yaml
 rename base/{ => nvidia}/nvidia-runtime.yaml (100%)

diff --git a/base/nvidia/nvidia-device-plugin.yaml b/base/nvidia/nvidia-device-plugin.yaml
new file mode 100644
index 0000000..fa80236
--- /dev/null
+++ b/base/nvidia/nvidia-device-plugin.yaml
@@ -0,0 +1,30 @@
+apiVersion: apps/v1
+kind: DaemonSet
+metadata:
+  name: nvidia-device-plugin-daemonset
+  namespace: kube-system
+spec:
+  selector:
+    matchLabels:
+      name: nvidia-device-plugin-ds
+  template:
+    metadata:
+      labels:
+        name: nvidia-device-plugin-ds
+    spec:
+      tolerations:
+      - key: nvidia.com/gpu
+        operator: Exists
+        effect: NoSchedule
+      containers:
+      - image: nvidia/k8s-device-plugin:1.0.0-beta6
+        name: nvidia-device-plugin-ctr
+        securityContext:
+          allowPrivilegeEscalation: false
+        volumeMounts:
+        - name: device-plugin
+          mountPath: /var/lib/kubelet/device-plugins
+      volumes:
+      - name: device-plugin
+        hostPath:
+          path: /var/lib/kubelet/device-plugins
diff --git a/base/nvidia-runtime.yaml b/base/nvidia/nvidia-runtime.yaml
similarity index 100%
rename from base/nvidia-runtime.yaml
rename to base/nvidia/nvidia-runtime.yaml