メインコンテンツへスキップ
見出し画像

How to install the TensorRT on ComfyUI

    先日、Syable Diffusion ComfyUIの作者comfyanonymous氏が、ComfyUI用のTensorRT拡張機能を発表しました。
    https://github.com/comfyanonymous/ComfyUI_TensorRT

    TensorRT高速化技術は、既にA1111が先行して拡張機能を実装していましたが、これでComyUIにおいてもTensorRT技術を使用できる環境が整ったことになります。

    今回は、そのインストール方法について解説したいと思います。

    尚、ComfyUIのインストール環境は、Windows上で一般的なportable形式で使用している前提で以下、解説していきます。

    まず下図のように、拡張機能のインストール先であるcustom_nodesフォルダにパスを通した形でpowershellを起動します。

    画像

    次に下図のように、
    git clone https://github.com/comfyanonymous/ComfyUI_TensorRT
    と入力して実行します。これによって拡張機能ファイルがダウンロードされます。

    画像

    次に下図のように、ComfyUI_windows_portableフォルダにパスを移動させます。ここから、TensorRTを動作させる為に必要なPythonライブラリーをインストールしていきます。

    画像

    次に下図のように、cudnnをインストールします。以下は、2025年9月11日時点の最新です。

    python_embeded\python.exe -m pip install nvidia-cudnn-cu12==9.15.0.57 --no-cache-dir

    このcudnn-cu12のインストールは公式の解説では不要の様ですが、A1111に対してTensorRTをインストールする際には必要な手順なので、念のために実行しておきます。
    (この手順を省略しても動作する可能性はあります)

    次に下図のように、入力して実行してください。これが、TensorRT本体のライブラリーになります。

    python_embeded\python.exe -m pip install --extra-index-url https://pypi.nvidia.com/ "tensorrt==10.12.0.36" --no-cache-dir

    次に、onnxruntime関係のライブラリーをインストールしていきます。

    公式の解説では、ComfyUIの場合はonnxだけインストールすれば良いようなのですが、ここでも念の為にA1111と同じライブラリーをインストールしていきます。

    まず、下図のように入力して実行してください。

    python_embeded\python.exe -m pip install importlib_metadata onnx polygraphy

    次に下図のように、onnxruntime-gpuをインストールします。

    python_embeded\python.exe -m pip install onnxruntime-gpu

    次に下図のように、requestsをインストールします。

    python_embeded\python.exe -m pip install requests

    以上で、ComfyUIの為のTensorRTのインストールは完了です。powershellを閉じて、ComfyUIを起動してください。

    モデルの作成方法と、TensorRTの適用方法に関しては公式の解説をご覧ください。
    (後日、追記するかもしれませんが)

    尚、下図のように\ComfyUI_windows_portable\ComfyUI\custom_nodes\ComfyUI_TensorRT\workflowsフォルダには、モデル作成及びTensorRTを適用する為の基本的なノードが用意されています。

    画像

    下図は、あらかじめ用意されているSD1.5用の基本的なワークフローの図です。私の環境では、かなり速度は上がりましたね。

    画像

    公式の解説を読みつつ、これらのノードを読み込めば、モデル作成も適用も問題なく実行できると思います。

    動作制限に関しては、やはりControlnetとLoRAはまだ使えないようです。

     
     

    GJL

     
     
    Road bike&Cycling Road Race, Respecting Mr.Greg LeMond, & Anti Doping, Football, Rock&Pops, History.

    あなたへのおすすめ