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
metadata:
name: immich-ml
namespace: media
namespace: default
spec:
strategy:
type: Recreate
@@ -16,9 +16,10 @@ spec:
labels:
app: immich-ml
spec:
runtimeClassName: nvidia
containers:
- 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:
- containerPort: 3003
env:
@@ -33,9 +34,9 @@ spec:
mountPath: /cache
resources:
requests:
gpu.intel.com/i915: "1"
nvidia.com/gpu: "1"
limits:
gpu.intel.com/i915: "1"
nvidia.com/gpu: "1"
volumes:
- name: model-cache
persistentVolumeClaim: