使用TensorRT-7.2.2.3转换onnx为trt文件时出现错误:Assertion failed: dims.nbDims == 4 || dims.nbDims == 5
[8] Assertion failed: dims.nbDims == 4 || dims.nbDims == 5
[07/28/2021-12:10:20] [E] Failed to parse onnx file
[07/28/2021-12:10:20] [E] Parsing model failed
[07/28/2021-12:10:20] [E] Engine creation failed
[07/28/2021-12:10:20] [E] Engine set up failed
经查阅资料,这里为版本错误,由于前面的 .pth 模型转换 onnx 文件时使用的 PyTorch 版本为1.8.0(与训练时 PyTorch 版本无关),需要降低到<=1.3.0 方可正常运行。重装版本为1.2.0后,运行成功: