From 43b2004a169d449748525845956afda95f3faf7b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Leon=20Michel=20Gori=C3=9Fen?= <leon.gorissen@llt.rwth-aachen.de> Date: Wed, 30 Oct 2024 17:08:34 +0100 Subject: [PATCH] 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. --- docker-compose.yaml | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/docker-compose.yaml b/docker-compose.yaml index 01f0f7c..0e6308a 100755 --- a/docker-compose.yaml +++ b/docker-compose.yaml @@ -138,7 +138,9 @@ services: resources: reservations: devices: - - capabilities: [gpu] + - driver: nvidia + device_ids: ['1'] + capabilities: [gpu] cpus: '10' memory: 30G ipc: host -- GitLab