I encountered an error of ImportError: No module named uff.
As I converted frozen graph to TensorRT engine with Jetpack3.3 without manaully install UFF.
I got another error as I converted frozen graph to TensorRT engine with Jetpack3.3 but manually install TensorRT 4 and uff.
The error message is as the below.
But I converted mobilenet_v1_0p25_128.plan with Jetpack3.3 and manaully installed TensorRT 3 and uff.
How to do tensorflow with tensorRT with Jetpack3.3 on TX2 exactly?
All the stpes I did are following the links below.
https://github.com/NVIDIA-Jetson/tf_to_trt_image_classification. (Jetpack 3.2 works)
error message :
nvidia@tegra-ubuntu:~/tf_to_trt_image_classification$ python scripts/convert_plan.py data/frozen_graphs/mobilenet_v1_0p25_128.pb data/plans/mobilenet_v1_0p25_128.plan input 128 128 MobilenetV1/Logits/SpatialSqueeze 1 0 float
Traceback (most recent call last):
File “scripts/convert_plan.py”, line 71, in
File “scripts/convert_plan.py”, line 22, in frozenToPlan
File “/usr/local/lib/python2.7/dist-packages/uff/converters/tensorflow/conversion_helpers.py”, line 149, in from_tensorflow_frozen_model
return from_tensorflow(graphdef, output_nodes, preprocessor, **kwargs)
File “/usr/local/lib/python2.7/dist-packages/uff/converters/tensorflow/conversion_helpers.py”, line 41, in from_tensorflow
https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/#python and click on the ‘TensoRT Python API’ link""".format(err))
ImportError: ERROR: Failed to import module (No module named graphsurgeon)
Please make sure you have graphsurgeon installed.
For installation instructions, see:
https://docs.nvidia.com/deeplearning/sdk/tensorrt-api/#python and click on the ‘TensoRT Python API’ link