---
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:v2.0.0-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