Add NVIDIA devices and GPU reservations to compose.yaml
Added NVIDIA device mappings and GPU resource reservations.
This commit is contained in:
parent
1392c2390a
commit
ff190b8c69
1 changed files with 10 additions and 0 deletions
10
compose.yaml
10
compose.yaml
|
|
@ -24,6 +24,16 @@ services:
|
|||
devices:
|
||||
- /dev/nvidia0:/dev/nvidia0
|
||||
- /dev/nvidiactl:/dev/nvidiactl
|
||||
- /dev/nvidia-uvm:/dev/nvidia-uvm
|
||||
- /dev/nvidia-uvm-tools:/dev/nvidia-uvm-tools
|
||||
- /dev/dri/renderD128:/dev/dri/renderD128
|
||||
deploy:
|
||||
resources:
|
||||
reservations:
|
||||
devices:
|
||||
- driver: nvidia
|
||||
count: all
|
||||
capabilities: [gpu]
|
||||
ports:
|
||||
- "8096:8096"
|
||||
networks:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue