web-dev-qa-db-ja.com

「uff」という名前のモジュールはありません

目標: tensorflow .pbモデルをtensorrtに変換するには

システム仕様: Ubundu 18.04 Cuda 10.0 TensorRT 5.1

インストール済みSudo apt-get install uff-converter-tf

Pythonでuffをインポートしようとしたときにエラーが発生しました:

ModuleNotFoundError:「uff」という名前のモジュールはありません

2
Sai Krishnadas

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-${version}.${os}.${Arch}-gnu.${cuda}.${cudnn}.tar.gzを解凍した結果として残りのTensorRTバイナリが含まれています。

いつものように、仮想環境にインストールするのが最適です。

4