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:
1 parent
a2f4a1799b
commit
7b31a531b4
2 files changed
+7
-7
No files matched your search
+1
-1
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user