Skip to content
Snippets Groups Projects
Commit 43b2004a authored by Leon Michel Gorißen's avatar Leon Michel Gorißen
Browse files

Updated `docker-compose.yaml` to specify GPU driver and device ID for NVIDIA.

- Added `driver: nvidia` and `device_ids: ['1']` under GPU configuration for more precise resource allocation.
- Kept existing CPU and memory reservations, ensuring compatibility with NVIDIA-based GPU setups.
parent a42958ec
No related branches found
No related tags found
No related merge requests found
......@@ -138,7 +138,9 @@ services:
resources:
reservations:
devices:
- capabilities: [gpu]
- driver: nvidia
device_ids: ['1']
capabilities: [gpu]
cpus: '10'
memory: 30G
ipc: host
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment