docker gpu #102

abdeladim-s · Jan 21, 2024 · 3e8714b · 3e8714b
1 parent cf71a44
commit 3e8714b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -4,4 +4,10 @@ services:
   subsai-webui:
     build:
       context: .
-      dockerfile: Dockerfile
+      dockerfile: Dockerfile
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              capabilities: [ gpu ]