From dcaed0e90ffd43a941d7bf35d6cf301c6ae2888c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20B=C3=A9dard-Couture?= Date: Sun, 6 Oct 2024 17:33:03 -0400 Subject: [PATCH] Log correct tensorRT version when debugging (#14182) --- frigate/detectors/plugins/tensorrt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index 37936e191..380fa9107 100644 --- a/frigate/detectors/plugins/tensorrt.py +++ b/frigate/detectors/plugins/tensorrt.py @@ -255,7 +255,7 @@ class TensorRtDetector(DetectionApi): raise RuntimeError("fail to allocate CUDA resources") from e logger.debug("TensorRT loaded. Input shape is %s", self.input_shape) - logger.debug("TensorRT version is %s", trt.__version__[0]) + logger.debug("TensorRT version is %s", TRT_VERSION) def __del__(self): """Free CUDA memories."""