概要
- JITでプログラム実行の際にコードが合理化され、学習が少しだけ速くなるライブラリ。
- 以前はLinux専用だったが、2025年頃Windows版が登場した。
- ComfyUIやsd-scriptsなどで導入できる
- ほかの高速化ライブラリを使用する上で必須となる。下記は一例
- torch.compile
- flash-attn
- SageAttention
インストール方法
triton導入
Windows版
triton-windows
導入済みのtorchバージョンを調べて、
そのtorchのバージョンに対応するtritonをインストールする
pip install triton-windows==XX.XX.XX XXにはVer.を記入。
Lunux版
(本来こちらが先に開発された事以外、筆者はよく知りません)