diff --git a/docker-compose-dev.yml b/docker-compose-dev.yml
index 57706a8..4df075c 100644
--- a/docker-compose-dev.yml
+++ b/docker-compose-dev.yml
@@ -59,9 +59,13 @@ services:
       - ./containers:/app
       - ./results:/results
       - ./logs:/logs
-    runtime: nvidia
-    devices:
-      - "/dev/nvidia0:/dev/nvidia0"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
 
   database:
     image: postgres:latest
diff --git a/docker-compose.yml b/docker-compose.yml
index e7d13eb..e998a49 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -43,8 +43,13 @@ services:
     volumes:
       - ./results:/results
       - ./logs:/logs
-    devices:
-      - "/dev/nvidia0:/dev/nvidia0"
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
 
   database:
     image: postgres:latest