use nvidia gpu

This commit is contained in:
2025-10-25 15:04:07 +05:30
parent 44ce9c58d3
commit 84ec7ada80

View File

@@ -3,7 +3,7 @@ apiVersion: apps/v1
kind: Deployment kind: Deployment
metadata: metadata:
name: immich-ml name: immich-ml
namespace: media namespace: default
spec: spec:
strategy: strategy:
type: Recreate type: Recreate
@@ -16,9 +16,10 @@ spec:
labels: labels:
app: immich-ml app: immich-ml
spec: spec:
runtimeClassName: nvidia
containers: containers:
- name: immich-machine-learning - name: immich-machine-learning
image: ghcr.io/immich-app/immich-machine-learning:v2.1.0-openvino image: ghcr.io/immich-app/immich-machine-learning:v2.1.0-cuda
ports: ports:
- containerPort: 3003 - containerPort: 3003
env: env:
@@ -33,9 +34,9 @@ spec:
mountPath: /cache mountPath: /cache
resources: resources:
requests: requests:
gpu.intel.com/i915: "1" nvidia.com/gpu: "1"
limits: limits:
gpu.intel.com/i915: "1" nvidia.com/gpu: "1"
volumes: volumes:
- name: model-cache - name: model-cache
persistentVolumeClaim: persistentVolumeClaim: