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

Add nvidia devices to compose file

Browse source
This commit is contained in:
Erik Stambaugh 2024-02-20 14:35:32 -08:00
parent e274ab10ed
commit d96897cb34
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
docker-compose.yaml
View file

@ -14,6 +14,12 @@ services:
- ./jupyter/overrides.json:/opt/conda/share/jupyter/lab/settings/overrides.json
environment:
- JUPYTER_TOKEN=12345
devices:
- /dev/nvidia-uvm
- /dev/nvidia-uvm-tools
- /dev/nvidia-modeset
- /dev/nvidiactl
- /dev/nvidia0
deploy:
resources:
reservations: