diff --git a/docker-compose.yaml b/docker-compose.yaml
index f85ec8181..da9619511 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -30,4 +30,5 @@ services:
         reservations:
           devices:
             - driver: nvidia
+              capabilities: [gpu]
               device_ids: ['all']