diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 50a19662109..c58cf8bf397 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -1,6 +1,12 @@
 services:
   stable-diffusion-webui:
-    runtime: nvidia
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities: [gpu]
     image: emsi/stable-diffusion-webui
     build:
       context: ../