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

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

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