No module named 'uff'

user_9222818 0
Viewed 2367

Goal : To convert tensorflow .pb model to tensorrt

System specs : Ubundu 18.04 Cuda 10.0 TensorRT 5.1

Even Installed sudo apt-get install uff-converter-tf

Error obtained when trying to import uff in Python:

ModuleNotFoundError: No module named 'uff'

uff tensorrt tensorflow python
1 Answers

Installing uff-converter-tf is not enough, you will need to install Python UFF wheel to be able to use it. Here you will find complete installation guide.

If using Python 2.7:

pip2 install tensorrt-*-cp27-none-linux_x86_64.whl

If using Python 3.x:

pip3 install tensorrt-*-cp3x-none-linux_x86_64.whl

You will find these wheels with rest of TensorRT binaries that were resulted from unpacking TensorRT-${version}.${os}.${arch}-gnu.${cuda}.${cudnn}.tar.gz.

As always, it's best if installed in virtual environment.

user_13979460
Related