diff --git a/docker-compose.yml b/docker-compose.yml
index 403c4db..a5dd9fa 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -3,6 +3,14 @@ version: "3.8"
 
 services:
   steam-headless:
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: 1
+              capabilities:
+                - gpu
     image: josh5/steam-headless:latest
     env_file:
       - stack.env
@@ -16,9 +24,6 @@ services:
         soft: 1024
         hard: 524288
 
-    # GPU PASSTHROUGH
-    runtime: nvidia
-
     # NETWORK:
     ## NOTE:  With this configuration, if we do not use the host network, then physical device input
     ##        is not possible and your USB connected controllers will not work in steam games.