Merge pull request #1 from steveh250/steveh250-patch-1
Updated GPU resource reservations to docker-compose.gpu.yml
This commit is contained in:
commit
68d56bb094
1 changed files with 7 additions and 2 deletions
|
|
@ -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]
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue