Model onnx trt engine generation process report different results compared between two PCs

AI & Data Science Deep Learning (Training & Inference) TensorRT
6

Hello,
The original onnx model cannot built to TRT engine.
I’m getting the following error:

> TRT - ERROR
[shuffleNode.cpp::nvinfer1::builder::ShuffleNode::symbolicExecute::392] Error Code 4: Internal Error (Reshape_66: IShuffleLayer applied to shape tensor must have 0 or 1 reshape dimensions: dimensions were [-1,2])
TRT - ERROR
ModelImporter.cpp:773: While parsing node number 76 [Pad → “143”]:
TRT - ERROR
ModelImporter.cpp:774: — Begin node —
TRT - ERROR
ModelImporter.cpp:775: input: “120”
input: “142”
output: “143”
name: “Pad_76”
op_type: “Pad”
attribute {

  • name: “mode”*
  • s: “reflect”*
  • type: STRING*
    }
    TRT - ERROR
    ModelImporter.cpp:776: — End node —
    TRT - ERROR
    ModelImporter.cpp:779: ERROR: ModelImporter.cpp:180 In function parseGraph:
    [6] Invalid Node - Pad_76
    [shuffleNode.cpp::nvinfer1::builder::ShuffleNode::symbolicExecute::392] Error Code 4: Internal Error (Reshape_66: IShuffleLayer applied to shape tensor must have 0 or 1 reshape dimensions: dimensions were [-1,2])

Attached is the original onnx model:
model.onnx (3.1 MB)

In order to avoid this error, I shall first run the Polygraphy on it and only then it successfully built to TRT engine but only on PC#1.
When I’m using the same setup (Same onnx* packages versions) on PC#2 and copy the folded model from PC#1 to PC#2 I’m not able to build the TRT engine on PC#2.

If I tried to fold the original model on PC#2 the problem described here is rasied:
https://forums.developer.nvidia.com/t/two-machines-with-very-similar-sw-stack-but-different-gpus-generate-different-folded-model-using-the-polygraphy-tool-on-the-same-model-onnx-input/217850

That’s why I copy the folded model from PC#1 to PC#2.

Please adivse,

Thanks