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

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

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