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:
|
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