diff --git a/docker-compose.gpu.yml b/docker-compose.gpu.yml
index 4496c4ac..b849cd3b 100644
--- a/docker-compose.gpu.yml
+++ b/docker-compose.gpu.yml
@@ -3,5 +3,10 @@ services:
     environment:
       - NVIDIA_DRIVER_CAPABILITIES=compute,utility
       - NVIDIA_VISIBLE_DEVICES=all
-    gpus: all
-
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]