diff --git a/docker-compose.yml b/docker-compose.yml
index 38ad63c..1126fe3 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -8,6 +8,8 @@ services:
       PORT: "12345"
       MODEL_DIR: "/app/models"
       DATA_DIR: "/data"
+      NVIDIA_VISIBLE_DEVICE: ALL
+    runtime: nvidia
     volumes:
       - ./server:/app/server
       - ./models:/app/models