Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats 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."""