diff --git a/cluster/apps/immich/immich-ml.yml b/cluster/apps/immich/immich-ml.yml new file mode 100644 index 0000000..83ab67c --- /dev/null +++ b/cluster/apps/immich/immich-ml.yml @@ -0,0 +1,43 @@ +--- +apiVersion: apps/v1 +kind: Deployment +metadata: + name: immich-ml + namespace: default +spec: + strategy: + type: Recreate + replicas: 1 + selector: + matchLabels: + app: immich-ml + template: + metadata: + labels: + app: immich-ml + spec: + runtimeClassName: nvidia + containers: + - name: immich-machine-learning + image: ghcr.io/immich-app/immich-machine-learning:v1.139.2-cuda + ports: + - containerPort: 3003 + env: + - name: REDIS_HOSTNAME + value: "immich-redis-service" + - name: NVIDIA_VISIBLE_DEVICES + value: "all" + - name: MACHINE_LEARNING_DEVICE_IDS + value: "0" + volumeMounts: + - name: model-cache + mountPath: /cache + resources: + requests: + nvidia.com/gpu: "1" + limits: + nvidia.com/gpu: "1" + volumes: + - name: model-cache + persistentVolumeClaim: + claimName: immich-cache-ceph diff --git a/cluster/apps/immich/immich-svc.yml b/cluster/apps/immich/immich-svc.yml index ffd293f..08098f4 100644 --- a/cluster/apps/immich/immich-svc.yml +++ b/cluster/apps/immich/immich-svc.yml @@ -21,7 +21,7 @@ metadata: spec: type: LoadBalancer selector: - app: immich-app + app: immich-ml ports: - port: 3333 targetPort: 3003 diff --git a/cluster/apps/immich/immich.yml b/cluster/apps/immich/immich.yml index 1095ddb..205be97 100644 --- a/cluster/apps/immich/immich.yml +++ b/cluster/apps/immich/immich.yml @@ -16,7 +16,6 @@ spec: labels: app: immich-app spec: - runtimeClassName: nvidia initContainers: - name: wait-for-redis image: busybox @@ -62,29 +61,7 @@ spec: volumeMounts: - mountPath: /usr/src/app/upload name: pictures - - name: immich-machine-learning - image: ghcr.io/immich-app/immich-machine-learning:v1.139.2-cuda - ports: - - containerPort: 3003 - env: - - name: REDIS_HOSTNAME - value: "immich-redis-service" - - name: NVIDIA_VISIBLE_DEVICES - value: "all" - - name: MACHINE_LEARNING_DEVICE_IDS - value: "0" - volumeMounts: - - name: model-cache - mountPath: /cache - resources: - requests: - nvidia.com/gpu: 1 - limits: - nvidia.com/gpu: 1 volumes: - name: pictures persistentVolumeClaim: claimName: immich-pictures-pvc - - name: model-cache - persistentVolumeClaim: - claimName: immich-cache-ceph