From 3e8714b148246dc06ab830bd981a01215ca14a68 Mon Sep 17 00:00:00 2001
From: su <sadiki.abdeladim@gmail.com>
Date: Sat, 20 Jan 2024 21:11:35 -0500
Subject: [PATCH] docker gpu #102

---
 docker-compose.yml | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/docker-compose.yml b/docker-compose.yml
index 3ca5072..c3a88bb 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -4,4 +4,10 @@ services:
   subsai-webui:
     build:
       context: .
-      dockerfile: Dockerfile
\ No newline at end of file
+      dockerfile: Dockerfile
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              capabilities: [ gpu ]
\ No newline at end of file