23 lines
606 B
YAML
23 lines
606 B
YAML
# Configurations for hardware-accelerated machine learning
|
|
|
|
# If using Unraid or another platform that doesn't allow multiple Compose files,
|
|
# you can inline the config for a backend by copying its contents
|
|
# into the immich-machine-learning service in the docker-compose.yml file.
|
|
|
|
# See https://immich.app/docs/features/ml-hardware-acceleration for info on usage.
|
|
|
|
services:
|
|
cpu: {}
|
|
|
|
cuda:
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: cdi
|
|
device_ids:
|
|
- nvidia.com/gpu=all
|
|
capabilities:
|
|
- gpu
|
|
|