fix(build&compose): Added dependencies & resource access

2025-11-11 00:22:25 +01:00
parent 46be4bb4b1
commit b3b34fa543
2 changed files with 27 additions and 9 deletions
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -5,7 +5,15 @@ services:
    restart: unless-stopped
    ports:
      - "8000:8000"
-    volumes:
-      - /sys/class/drm/:/sys/class/drm/:ro
-      - /usr/bin/nvidia-smi:/usr/bin/nvidia-smi:ro
-      - /usr/bin/rocm-smi:/usr/bin/rocm-smi:ro
+    devices:
+      - /dev/kfd
+      - /dev/dri
+    security_opt:
+      - seccomp:unconfined
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - capabilities: [gpu]
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all