Added configs to enable machine learning gpu feature

This commit is contained in:
Daniel Cardozo
2025-10-28 12:56:01 +01:00
parent 573c1b3d41
commit 4bd41ccf77

View File

@@ -4,7 +4,7 @@ metadata:
name: immich-machine-learning name: immich-machine-learning
namespace: immich namespace: immich
annotations: annotations:
reloader.stakater.com/auto: 'true' reloader.stakater.com/auto: "true"
labels: labels:
app: immich-machine-learning app: immich-machine-learning
spec: spec:
@@ -19,7 +19,7 @@ spec:
automountServiceAccountToken: false automountServiceAccountToken: false
containers: containers:
- name: immich-machine-learning - name: immich-machine-learning
image: ghcr.io/immich-app/immich-machine-learning:v2.1.0 # {"$imagepolicy": "immich:immich-machine-learning"} image: ghcr.io/immich-app/immich-machine-learning:v2.1.0-openvino # {"$imagepolicy": "immich:immich-machine-learning"}
imagePullPolicy: IfNotPresent imagePullPolicy: IfNotPresent
resources: resources:
limits: limits:
@@ -52,6 +52,14 @@ spec:
name: immich-db-app name: immich-db-app
key: password key: password
optional: false optional: false
# Check this ENVs maybe it works also for other ML models
- name: MACHINE_LEARNING_PRELOAD__CLIP
value: ViT-B-32__openai
- name: MACHINE_LEARNING_PRELOAD__FACIAL_RECOGNITION
value: buffalo_l
# May not be needed
- name: MACHINE_LEARNING_EXECUTION_PROVIDERS
value: "OpenVINOExecutionProvider,CPUExecutionProvider"
envFrom: envFrom:
- configMapRef: - configMapRef:
name: immich name: immich
@@ -72,12 +80,12 @@ spec:
securityContext: securityContext:
privileged: true # for intel quick sync privileged: true # for intel quick sync
seccompProfile: seccompProfile:
type: 'RuntimeDefault' type: "RuntimeDefault"
volumeMounts: volumeMounts:
- name: immich-machine-learning - name: immich-machine-learning
mountPath: /cache mountPath: /cache
nodeSelector: nodeSelector:
intel.feature.node.kubernetes.io/gpu: 'true' intel.feature.node.kubernetes.io/gpu: "true"
volumes: volumes:
- name: immich-machine-learning - name: immich-machine-learning
persistentVolumeClaim: persistentVolumeClaim: