From 9c7527a24ef07dccd0529bc9f0d9069de35b44af Mon Sep 17 00:00:00 2001
From: vladd-bit <vlad.a.dinu@gmail.com>
Date: Wed, 17 Apr 2024 12:22:49 +0100
Subject: [PATCH] Added extra gpu settings to compose file.

---
 docker/docker-compose.yml | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index dd09884..9449286 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -27,6 +27,15 @@ services:
       - "5555:5000"
     networks:
       - cognet
+    
+    # uncomment this only when you have gpu access  
+    #deploy:
+    #  resources:
+    #    reservations:
+    #      devices:
+    #        - driver: nvidia
+    #          device_ids: ['0']
+    #          capabilities: ["gpu", "utility", "compute", "video"]
       
 networks:
   cognet: