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