Add tensorrt>=7.0.0 checks (#6193)

* Add `tensorrt>=7.0.0` checks

* Update export.py

* Update common.py

* Update export.py
This commit is contained in:
Glenn Jocher authored and GitHub committed 2022-01-04 13:39:13 -08:00
1 parent a2f4a1799b
commit 7b31a531b4
2 files changed
+7 -7

No files matched your search

+1 -1
View File
@@ -337,7 +337,7 @@ class DetectMultiBackend(nn.Module):
elif engine: # TensorRT
LOGGER.info(f'Loading {w} for TensorRT inference...')
import tensorrt as trt # https://developer.nvidia.com/nvidia-tensorrt-download
check_version(trt.__version__, '8.0.0', verbose=True) # version requirement
check_version(trt.__version__, '7.0.0', hard=True) # require tensorrt>=7.0.0
Binding = namedtuple('Binding', ('name', 'dtype', 'shape', 'data', 'ptr'))
logger = trt.Logger(trt.Logger.INFO)
with open(w, 'rb') as f, trt.Runtime(logger) as runtime: