From 40fc5e34090d00a73b8677caa0f38f0153a9510c Mon Sep 17 00:00:00 2001
From: marcus <marcus@freedoh.net>
Date: Sat, 20 Jan 2024 08:37:07 +0000
Subject: [PATCH] Update docker-compose.yml

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

diff --git a/docker-compose.yml b/docker-compose.yml
index b1a1bd2..4ae4da5 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,7 +1,8 @@
 version: "3"
 services:
   frigate:
-    image: ghcr.io/blakeblackshear/frigate:stable
+    # image: ghcr.io/blakeblackshear/frigate:stable
+    image: ghcr.io/blakeblackshear/frigate:stable-tensorrt
     container_name: frigate
     # user: vscode
     privileged: true
@@ -17,6 +18,13 @@ services:
           size: 1000000000
       - /dev/bus/usb:/dev/bus/usb
       - /dev/dri:/dev/dri # for intel hwaccel, needs to be updated for your hardware
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
     ports:
       - "5935:1935"
       - "5000:5000"