From 1bd26a082cbfbce61e43235a9474a73571c3190f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20B=C3=A9dard-Couture?= Date: Sat, 18 May 2024 15:57:59 -0400 Subject: [PATCH] Update frigate/detectors/plugins/tensorrt.py Co-authored-by: Nicolas Mowen --- frigate/detectors/plugins/tensorrt.py | 1 + 1 file changed, 1 insertion(+) diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 650212292..f1c3d2511 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -6,6 +6,7 @@ import numpy as np try: import tensorrt as trt from cuda import cuda + TRT_VERSION = int(trt.__version__[0 : trt.__version__.find(".")]) TRT_SUPPORT = True