apps/frigate-rockchip
1
0
Fork 0
mirror of https://github.com/MarcA711/frigate-rockchip.git synced 2025-10-07 01:32:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Log correct tensorRT version when debugging (#14182)

Browse Source
This commit is contained in:
Rémi Bédard-Couture
2024-10-06 17:33:03 -04:00
committed by GitHub
parent cae304e07f
commit dcaed0e90f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/detectors/plugins/tensorrt.py
View File

@@ -255,7 +255,7 @@ class TensorRtDetector(DetectionApi):
raise RuntimeError("fail to allocate CUDA resources") from e raise RuntimeError("fail to allocate CUDA resources") from e
logger.debug("TensorRT loaded. Input shape is %s", self.input_shape) 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): def __del__(self):
"""Free CUDA memories.""" """Free CUDA memories."""