44 lines
987 B
YAML
44 lines
987 B
YAML
---
|
|
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.1-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
|