I have an Ubuntu 16.04 system, GTX 1060, CUDA 10.0, TensorRT.
I am following the basic installation instructions.
uff converter is installed (appears to be correct):
sudo apt-get install uff-converter-tf
Reading package lists… Done
Building dependency tree
Reading state information… Done
uff-converter-tf is already the newest version (5.1.5-1+cuda10.0).
when I attempt to convert a frozen graph (python3.5 using virtualenv):
/usr/lib/python3.5/dist-packages/uff/bin$ python convert_to_uff.py tflite_graph.pb
Traceback (most recent call last):
File “convert_to_uff.py”, line 15, in
import uff
ImportError: No module named ‘uff’
What am I missing?
I don’t see any uff module on my computer:
find /usr -name uff*.*
/usr/share/doc/tensorrt/python/uff/uff.html
/usr/share/doc/tensorrt/python/_sources/uff/uff.rst.txt
/usr/src/tensorrt/samples/python/introductory_parser_samples/uff_resnet50.py
/usr/lib/python2.7/dist-packages/uff/model/uff_pb2.py
/usr/lib/python2.7/dist-packages/uff-0.6.3.dist-info
/usr/lib/python3.5/dist-packages/uff/model/uff_pb2.py
/usr/lib/python3.5/dist-packages/uff-0.6.3.dist-info