環境
Windows 11 Pro
CPU：Core i9-13900KF
GPU：GeForce RTX 4090 24GB
メモリ：64GB
Python：3.10.11
CUDA：11.8
cuDNN：8.9.7
PyTorch：2.1.1
StreamDiffusionインストール
これ通りに進める。
問題なく動かせたけど、conda環境でもvenv環境でもTensorRTをONにするための
python -m streamdiffusion.tools.install-tensorrt
で
python.exe エントリポイントが見つかりません
.conda\envs\NAME\lib\site-packages\nvidia\cudnn\bin\cudnn_cnn_infer64_8.dll
のようなエラーが出る。
CUDAとかのパスとかはちゃんと通ってたので困ってたところ、
「StreamDiffusion\venv\Lib\site-packages\nvidia\」にある「cudnn」ディレクトリの名前を、「cudnn.bak」等に変更してください。これによって参照されるcuDNNの場所が変更されて、エラーのダイアログが出現しなくなると思います。
ありがとうございます。
出たエラー集
結果的にフォルダ名の変更で全部解決できたのですが、記録として以下を残しておきます。
「python.exe エントリポイントが見つかりません」と表示されていたけどtensorrtをオンでやってみた
ModuleNotFoundError: No module named 'polygraphy' Acceleration has failed. Falling back to normal mode.
→polygraphyをインストールしてみるか。
pip install polygraphy==0.47.1とするとそんなバージョンないよというエラー。
ERROR: Could not find a version that satisfies the requirement polygraphy==0.47.1 (from versions: 0.0.1.dev4, 0.0.1.dev5, 0.48.1, 0.49.0)
ERROR: No matching distribution found for polygraphy==0.47.1
じゃあ0.48.1か0.49.0をインストールすると
AttributeError: module 'polygraphy.backend.trt.util' has no attribute 'get_bindings_per_profile'
→polygraphyのバージョンが合ってないと起こるらしいエラー。
それで、READMEにある
python -m streamdiffusion.tools.install-tensorrt
だと、polygraphyの0.47.1がインストールされるみたい。
CUDAの別バージョンをインストールし直そうとしたりしたけど必要なくてよかった。
終わりに
エラーを検索してこの記事に辿り着いた人に幸あれ。