mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-02-11 05:35:25 +03:00
Update frigate/detectors/plugins/tensorrt.py
Co-authored-by: Nicolas Mowen <nickmowen213@gmail.com>
This commit is contained in:
parent
be84fe8a5e
commit
1bd26a082c
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user