From 11fbd645f7aee22cfb54b59555558d538e51e385 Mon Sep 17 00:00:00 2001
From: Nikhil Shrestha <nikhil.shrestha1995@gmail.com>
Date: Sun, 5 May 2024 08:22:00 +0545
Subject: [PATCH] enable gpu all

---
 docker-compose.yaml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker-compose.yaml b/docker-compose.yaml
index 193add27..cacfb19b 100644
--- a/docker-compose.yaml
+++ b/docker-compose.yaml
@@ -46,8 +46,8 @@ services:
         reservations:
           devices:
             - driver: nvidia
-              count: 1
-              capabilities: [ gpu ]
+              capabilities: [ "gpu" ]
+              count: all
 
   db:
     image: postgres:15-alpine