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