diff --git a/kubernetes/apps/media/immich/app/machine-learning/helmrelease.yaml b/kubernetes/apps/media/immich/app/machine-learning/helmrelease.yaml
index 0f7ef1cb..ffe0409e 100644
--- a/kubernetes/apps/media/immich/app/machine-learning/helmrelease.yaml
+++ b/kubernetes/apps/media/immich/app/machine-learning/helmrelease.yaml
@@ -62,6 +62,7 @@ spec:
     pod:
       nodeSelector:
         nvidia.com/gpu.present: "true"
+      runtimeClassName: nvidia
     service:
       app:
         controller: immich-machine-learning
diff --git a/kubernetes/apps/media/immich/app/microservices/helmrelease.yaml b/kubernetes/apps/media/immich/app/microservices/helmrelease.yaml
index 2bf3a799..a5a97bcc 100644
--- a/kubernetes/apps/media/immich/app/microservices/helmrelease.yaml
+++ b/kubernetes/apps/media/immich/app/microservices/helmrelease.yaml
@@ -65,6 +65,7 @@ spec:
     pod:
       nodeSelector:
         nvidia.com/gpu.present: "true"
+      runtimeClassName: nvidia
     service:
       app:
         controller: immich-microservices