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 !!!