From dcb62c838d8d91376496ad647a0772ebabb902e1 Mon Sep 17 00:00:00 2001
From: Jack Jackson <scubbojj@gmail.com>
Date: Sat, 8 Feb 2025 14:13:49 -0800
Subject: [PATCH] Use nvidia runtimeClassName

---
 charts/jellyfin/templates/deployment.yaml | 8 +++-----
 charts/jellyfin/values.yaml               | 2 ++
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/charts/jellyfin/templates/deployment.yaml b/charts/jellyfin/templates/deployment.yaml
index dd71c24..1cacc56 100644
--- a/charts/jellyfin/templates/deployment.yaml
+++ b/charts/jellyfin/templates/deployment.yaml
@@ -26,6 +26,9 @@ spec:
       {{- end }}
       securityContext:
         {{- toYaml .Values.podSecurityContext | nindent 8 }}
+      {{- if .Values.runtimeClassName }}
+      runtimeClassName: {{ .Values.runtimeClassName }}
+      {{- end }}
       containers:
         - name: {{ .Chart.Name }}
           securityContext:
@@ -42,8 +45,6 @@ spec:
               name: jf-config
             - mountPath: /cache
               name: jf-cache
-            - mountPath: /dev/dri
-              name: dri
       tolerations:
         - key: face
           operator: Equal
@@ -59,9 +60,6 @@ spec:
         - name: jf-cache
           persistentVolumeClaim:
             claimName: jf-cache-pvc
-        - name: dri
-          hostPath:
-            path: /dev/dri
       {{- with .Values.nodeSelector }}
       nodeSelector:
         {{- toYaml . | nindent 8 }}
diff --git a/charts/jellyfin/values.yaml b/charts/jellyfin/values.yaml
index 58699ce..a7a824f 100644
--- a/charts/jellyfin/values.yaml
+++ b/charts/jellyfin/values.yaml
@@ -127,6 +127,8 @@ metricsImage:
   repository: gitea.scubbo.org/scubbo/jellyfin-library-count-prometheus-exporter
   tag: latest
 
+runtimeClassName: nvidia
+
 metrics:
   apiUrl: "http://jellyfin.avril"
   affinity: