Description
I use trtexec (TensorRT 8.5.0 from the nvidia NGC pytorch 22.09-py3 container) to convert an ONNX model to trt engine.
There is an issue when I compile the ONNX model. After several minute on a A100-40 GB GPU it show the error below :
[02/07/2023-11:12:18] [W] [TRT] Using kFASTER_DYNAMIC_SHAPES_0805 preview feature. [02/07/2023-11:13:44] [W] [TRT] Skipping tactic 0x0000000000000000 due to Myelin error: Autotuner: no tactics to implement operation [02/07/2023-11:13:44] [E] Error[10]: [optimizer.cpp::computeCosts::3712] Error Code 10: Internal Error (Could not find any implementation for node {ForeignNode[onnx::Gather_543...Transpose_3934 + Reshape_3943]}.) [02/07/2023-11:13:44] [E] Error[2]: [builder.cpp::buildSerializedNetwork::738] Error Code 2: Internal Error (Assertion engine != nullptr failed. )
Here is the command that I've used:
trtexec --onnx=/home/my_model.onnx --saveEngine=/home/my_model.onnx_trt --fp16 --inputIOFormats=fp16:chw --outputIOFormats=fp16:chw --workspace=40000 --minShapes=input_0:1x3x896x896 --optShapes=input_0:1x3x896x896 --maxShapes=input_0:1x3x896x896 --preview=+fasterDynamicShapes0805
I've tried with smaller workspace size, with and without the fasterDynamicShapes0805 preview features, and with/without dynamic inputs size. Do you have any idea on how to solve this issue?