use nvidia gpu
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user