[8] Assertion failed: dims. nbDims == 4 || dims. nbDims == 5

onnx turn trt There was a mistake ：

[04/22/2022-15:45:13] [W] [TRT] onnx2trt_utils.cpp:220: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
[04/22/2022-15:45:13] [E] [TRT] (Unnamed Layer* 371) [Shuffle]: at most one dimension may be inferred
ERROR: onnx2trt_utils.cpp:1517 In function scaleHelper:
[8] Assertion failed: dims.nbDims == 4 || dims.nbDims == 5
[04/22/2022-15:45:13] [E] Failed to parse onnx file
[04/22/2022-15:45:13] [E] Parsing model failed
[04/22/2022-15:45:13] [E] Engine creation failed
[04/22/2022-15:45:13] [E] Engine set up failed

according to   This article , Can solve , But it needs to be upgraded trt.

No upgrade trt The solution is , recompile libnvonnxparser.so, Replace trtexec Version used .

The specific steps are ：（ Assuming the current trt The version is 7.2.1.6, Installed in /usr/local/TensroRT7.2.1.6）

1、 download

git clone -b 7.2.1 https://github.com/onnx/onnx-tensorrt.git

2、 compile

cd onnx-tensorrt
git submodule update --init --recursive
mkdir build && cd build
cmake .. -DTENSORRT_ROOT=/usr/local/TensorRT-7.2.1.6 && make -j

3、 use ldd trtexec Look at what it depends on libnvonnxparser.so Where is the , Replace with the corresponding newly generated file .

Reference resources ：

Problem converting ONNX model to TensorRT 7.2.1 (worked in previous 7.1.3 release) · Issue #566 · onnx/onnx-tensorrt · GitHub