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: