用途:为 ComfyUI(尤其 WanVideoWrapper 等视频图)安装两个可选加速组件——SageAttention(量化注意力内核,通过 attention_mode=sageattn 或 --use-sage-attention 启用,支持的 NVIDIA GPU 上采样约快 20–40%)与 Triton(inductor torch.compile 所需的 GPU 内核编译器)。两者均与 torch、CUDA、Python 版本严格绑定,装错可能破坏 torch 环境。
主要能力与用法:优先使用 sdpa / 不编译的回退方案先跑通渲染,再询问是否安装加速。若 kitchen INT8 注意力可用,改用 --use-ck-attention 并跳过 SageAttention 安装。Windows 无官方构建,需用社区预编译 wheel:先用目标 Python 读取 torch/CUDA/Python 版本,再按 torch 版本选择 triton-windows 约束(如 torch 2.10 用 "triton-windows<3.7"),SageAttention 按 cu128/cu130 与 torch 小版本选 wheel,用完整 URL 安装。Linux 可直接 pip install -U triton 与 pip install sageattention。Mac 无 CUDA,不安装,始终用 sdpa/MPS。
验证与限制:安装后需确认 import triton、import sageattention 成功且 torch 仍可导入并可用 CUDA,ComfyUI 启动日志无相关报错。常见陷阱包括装错 Python、wheel 与 torch/CUDA 不匹配、Triton 缓存过期(可删除 ~/.triton)、Windows 缺 MSVC 或嵌入式 Python 缺头文件。若输出质量下降应回退 sdpa。Z-Image 在 --use-sage-attention 下异常,需改用 --use-pytorch-cross-attention。