Can I use TensorRT with Python APi on TX2?

I want to accelerate my network by TensorRT. I use Pytorch. Can I use TensorRT with Python APi on TX2?

As far as I know, TensorRT Python API is not supported on Jetson platform due to pyCUDA. Here is an alternative for your reference:

thank you

Hi,

Python API does support Jetstom platform from JetPack4.2.
You check our TensorRT samples for more information:

/usr/src/tensorrt/samples/python/yolov3_onnx/
/usr/src/tensorrt/samples/python/network_api_pytorch_mnist/

Thanks.

Very good news! I didn’t check the latest version. I’m sorry @ 2360121331.

Thank you !!!