how to use python api for network optimization using TensorRT on jetson TX2

Autonomous Machines Jetson & Embedded Systems Jetson TX2
1

Hi all:
I flashed my jetson TX2 board with new jetpack 3.2 developer preview. Previously I had jetpack 3.1 with TensorRT 2.1.
Now jetpack 3.2 has TensorRT 3. I know C++ API for tensorRT can be used by including the header NVinfer.h. I don’t know how to use Python Api for TensorRT which packages I need to import.
Can anyone please explain the workflow of using TensorRT Python Api on Jetson TX2?
Are there any sample examples available for tensorRT python Api for Jetson TX2?

2

Hi,

Python API is only available on x86 Linux platform, can’t be used on Jetson.

Here is the support information:
[url]Developer Guide :: NVIDIA Deep Learning TensorRT Documentation

Thanks.

3

Thanks, but i want to use python api on jetson. Is there any other way to do this?

4

Hi,

Currently, python API doesn’t support Jetson platform.
Please wait for our future release.

Thanks.

5

May i know when it will be released.

6

Hi,

Sorry for that we cannot disclose our schedule here.

Python API is not implemented in our next JetPack release but is in our plan.
Please pay attention to our announcement for an update.

Thanks.

7

Can I ask:
As the Jetpack 3.2 formal version released, Does python API support Jetson platform now?

8

Not yet. Python API is not available for Jetson currently.

If you want to convert a TensorFlow model into TensorRT, here is a good tutorial for your reference:
[url]https://github.com/NVIDIA-Jetson/tf_to_trt_image_classification[/url]

Thanks.

9

Thanks

10

I’ve created a TensorRT GoogLeNet example, in which I used Cython to wrap C++ code so that I could do TensorRT inferencing using python directly. Hope it helps.

The code was tested on Jetson Nano, but it should work on Jetson TX2 too.