diff --git a/frigate/detectors/plugins/tensorrt.py b/frigate/detectors/plugins/tensorrt.py index b362f0b0b..64b0849c7 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