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