nvidia nx onnx转trt模型报错

[2022-11-17 18:33:20][error][trt_builder.cpp:30]:NVInfer: …/builder/cudnnBuilderBlockChooser.cpp (117) - Assertion Error in buildMemGraph: 0 (mg.nodes[mg.regionIndices[outputRegion]].size == mg.nodes[mg.regionIndices[inputRegion]].size)

解决:

onnx_model, check = onnxsim.simplify(onnx_model)

转化简化的onnx时有问题,去掉这句。

你可能感兴趣的:(深度学习)