目標: tensorflow .pbモデルをtensorrtに変換するには
システム仕様: Ubundu 18.04 Cuda 10.0 TensorRT 5.1
インストール済みSudo apt-get install uff-converter-tf
Pythonでuff
をインポートしようとしたときにエラーが発生しました:
ModuleNotFoundError:「uff」という名前のモジュールはありません
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バイナリが含まれています。
いつものように、仮想環境にインストールするのが最適です。