Update compose.yaml: use GPU

DefangLabs · Sep 26, 2024 · 1af9610 · 1af9610
1 parent 49721d0
commit 1af9610
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/compose.yaml b/compose.yaml
@@ -19,8 +19,8 @@ services:
       resources:
         reservations:
           memory: 4G
-          # devices:
-          #   - capabilities: ["gpu"]
+          devices:
+            - capabilities: ["gpu"]
     healthcheck:
       test: ["CMD", "curl", "-f", "http://localhost:5000/"]
       interval: 30s