Нет модуля с именем 'uff'

Question

Цель: Преобразовать модель тензорного потока .pb в тензортр

Характеристики системы: Убунду 18.04 Cuda 10.0 TensorRT 5.1

Даже установлен sudo apt-get install uff-converter-tf

Ошибка при попытке импорта uff в Python:

ModuleNotFoundError: Нет модуля с именем 'uff'

Answer 1

Установка uff-converter-tf недостаточна, вам нужно установить колесо Python UFF, чтобы иметь возможность его использовать. Здесь вы найдете полное руководство по установке.

При использовании Python 2.7:

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

При использовании Python 3.x:

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

Вы найдете эти колеса с остальными двоичными файлами TensorRT, которые возникли в результате распаковки TensorRT-${version}.${os}.${arch}-gnu.${cuda}.${cudnn}.tar.gz.

Как всегда, лучше всего, если он установлен в виртуальной среде.