Log correct tensorRT version when debugging (blakeblackshear#14182)

ivanjx · Oct 6, 2024 · dcaed0e · dcaed0e
1 parent cae304e
commit dcaed0e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py
@@ -255,7 +255,7 @@ def __init__(self, detector_config: TensorRTDetectorConfig):
             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."""