Updated GPU resource reservations to docker-compose
Original version was failing on Ubuntu EC2 instance with GPU - error message was: 'validating /home/ubuntu/.openrag/tui/docker-compose.gpu.yml: services.openrag-backend Additional property gpus is not allowed Command exited with status 15
This commit is contained in:
parent
11b90f20bb
commit
cdec80288e
1 changed files with 7 additions and 2 deletions
|
|
@ -3,5 +3,10 @@ services:
|
||||||
environment:
|
environment:
|
||||||
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
||||||
- NVIDIA_VISIBLE_DEVICES=all
|
- NVIDIA_VISIBLE_DEVICES=all
|
||||||
gpus: all
|
deploy:
|
||||||
|
resources:
|
||||||
|
reservations:
|
||||||
|
devices:
|
||||||
|
- driver: nvidia
|
||||||
|
count: all
|
||||||
|
capabilities: [gpu]
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue