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"