Update frigate/detectors/plugins/tensorrt.py

Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com>
This commit is contained in:
Rémi Bédard-Couture 2024-05-18 15:57:59 -04:00 committed by GitHub
parent be84fe8a5e
commit 1bd26a082c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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