diff --git a/docker-compose.yml b/docker-compose.yml
index fd5bf70a..b2563790 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -11,13 +11,13 @@ services:
       - ./src:/app/src
       - ./saves:/app/saves
       - ${MODEL_DIR:-./models}:/models  # 使用默认值处理未定义的环境变量
-    # deploy:
-    #   resources:
-    #     reservations:
-    #       devices:
-    #         - driver: nvidia
-    #           device_ids: ['1']
-    #           capabilities: [gpu]
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              device_ids: ['1']
+              capabilities: [gpu]
     ports:
       - "5050:5050"
     networks: