Merge pull request #1 from steveh250/steveh250-patch-1

Updated GPU resource reservations to docker-compose.gpu.yml
This commit is contained in:
Steven Harris 2026-01-11 14:08:13 -08:00 committed by GitHub
commit 68d56bb094
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -3,5 +3,10 @@ services:
environment:
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
- NVIDIA_VISIBLE_DEVICES=all
gpus: all
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: all
capabilities: [gpu]