PyTorch_ONNX_TensorRT

PyTorch_ONNX_TensorRT copied to clipboard

trtexec --onnx=/home/guohao02/PyTorch_ONNX_TensorRT/model_128.onnx --explicitBatch ``` [10/16/2021-13:56:29] [I] === Model Options === [10/16/2021-13:56:29] [I] Format: ONNX [10/16/2021-13:56:29] [I] Model: /home/guohao02/PyTorch_ONNX_TensorRT/model_128.onnx [10/16/2021-13:56:29] [I] Output: [10/16/2021-13:56:29] [I] === Build Options === [10/16/2021-13:56:29] [I] Max...

guohao02

henbucuoshanghai

代码与 tensorrt 8.4 不兼容

1

我正在使用 tensorrt 8.4的环境运行代码，报错 'tensorrt.tensorrt.Builder' object has no attribute 'build_cuda_engine'，int8_mod, fp16_mode, int8_mode 在新的tensorrt中，使用config 接管了 builder的各种设置。

chenxinfeng4