Tensorflow to TensorRT UFF conversion on Windows

Robotics & Edge Computing Jetson & Embedded Systems Jetson TX2
1

Hi,

I am trying to convert a TF model to a TRT model to run on my Jetson TX2. On my Jetson, I have TensorRT version 3.0.4. I have two questions:

  1. Can I convert my TensorFlow model to UFF format directly on my windows machine? If so, how?

  2. If I cannot do it on my windows machine, but only on a linux machine, should I install the latest version of TensorRT on that machine (currently version 5), or should I install the same version of TensorRT that is currently on my Jetson (v3.0.4).

Thank you!
Roman

2

Hi,

I haven’t tested this on Windows, but I believe the UFF package is python only (no compiled components) and may work. Can you try downloading the TensorRT tar package from here

https://developer.nvidia.com/nvidia-tensorrt-download

I would download the version of TensorRT that matches what you have on your Jetson. Once you’ve downloaded and extracted the package you may be able to install the UFF package using pip. For example, for TensorRT 3.0.4 you would do

pip install TensorRT-3.0.4/uff/uff-0.2.0-py2.py3-none-any.whl

Please let me know if this works or if you run into any issues!

Thanks,
John

3

@Jaybdub, the UFF package successfully installed, however I have not had a chance to use it yet. Will update after I do…

Thank you!

@