erik/mlops
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update ulimits according to nvidia's recommendations

Browse source
This commit is contained in:
Erik Stambaugh 2024-03-02 14:51:50 -08:00
parent d1baec0f60
commit d3bd07dcc6
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
docker-compose.yaml
View file

@ -22,13 +22,17 @@ services:
- /dev/nvidia-modeset - /dev/nvidia-modeset
- /dev/nvidiactl - /dev/nvidiactl
- /dev/nvidia0 - /dev/nvidia0
ulimits:
memlock: -1
stack: 67108864
ipc: host
deploy: deploy:
resources: resources:
reservations: reservations:
devices: devices:
- driver: nvidia - driver: nvidia
count: 1 count: all
capabilities: capabilities: [gpu]
- gpu