TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

How to get the profile by python?

Open oreo-lp opened this issue 3 years ago • 1 comments

Description

I can get the profile by trtexce: ./trtexec --loadEngine=debug_fp16.trt --dumpProfile --shapes=input:1x3x512x512 --exportProfile=debug_profile How can I get the debug_profile by python when I convert onnx to trt engine by python?

oreo-lp avatar Aug 10 '22 07:08 oreo-lp

I haven't used it before but I guess https://docs.nvidia.com/deeplearning/tensorrt/api/python_api/infer/Core/ExecutionContext.html#tensorrt.IExecutionContext.report_to_profiler is the answer, @nvpohanh may know more about it.

zerollzeng avatar Aug 10 '22 14:08 zerollzeng

closing since no activity for more than 3 weeks, please reopen if you still have question, thanks!

ttyio avatar Dec 06 '22 02:12 ttyio