From a1a03f29c9b10e37ef9b46c3726d23bad28f4b97 Mon Sep 17 00:00:00 2001
From: Elias Stepanik <40958815+eliasstepanik@users.noreply.github.com>
Date: Wed, 22 Feb 2023 08:09:33 +0100
Subject: [PATCH] Update docker-compose.yml

---
 docker-compose.yml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index f93ff34..dfd6bc6 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,9 +11,9 @@ x-base_service: &base_service
       resources:
         reservations:
           devices:
-              - driver: nvidia
-                device_ids: ['0']
-                capabilities: [gpu]
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
 volumes:
   data:
   output: