Triton & SageAttention 一键安装指南（Windows 平台）

为了加速 ComfyUI 中的图片和视频生成速度，开发者推出了 Triton 和 SageAttention 的一键安装程序。Triton 是一种用于高性能计算的编译器，而 SageAttention 是一种优化注意力机制的库。两者结合可以显著提升生成任务的速度。然而，在 Windows 上安装这些工具通常较为复杂，因此一键安装程序的出现极大地简化了安装流程。

• 来源：https://civitai.com/articles/12248
• 备份：https://www.123865.com/s/hyQyTd-oUhDv 提取码:jUrr

一键安装程序简介

一键安装程序通过预编译的包和脚本，帮助用户在 Windows 上快速安装 Triton 和 SageAttention。安装完成后，用户可以在 ComfyUI 中更高效地进行图片和视频生成任务。

安装步骤

1. 下载安装包
   • 下载附件中的 zip 文件，并解压其中的 bat 文件。
   • 将 bat 文件放入以下路径之一：
     • ComfyUI 桌面路径：ComfyUI\.venv\
     • Pinokio 路径：pinokio\api\comfy.git\app\env\
     • StableMatrix 路径：StabilityMatrix\Assets\
2. 运行安装脚本
   • 双击运行 bat 文件，开始安装过程。

系统要求

• 硬件要求
  • 英伟达RTX 30 系或 RTX 40 系显卡
  • 至少 8GB 显存（建议更高配置）
• 软件要求
  • PyTorch：版本 >= 2.6
  • CUDA Toolkit：版本 12.x
    • RTX 30/40 系显卡：CUDA Toolkit 12.6.3
    • RTX 50 系显卡：CUDA Toolkit 12.1
  • Python：版本 3.12（ComfyUI 环境）
  • ComfyUI：支持 Wan / Hunyuan / Skyreels 工作流

测试环境

• 操作系统：Windows 11
• CUDA 工具包：12.6.3
• ComfyUI：Python 3.12
• 显卡：
  • RTX 3090 24GB
  • RTX 4060 8GB

自定义安装

如果您需要更改 Triton 的版本，请打开 bat 文件并修改相应的下载地址。例如，对于 RTX 50 系显卡，可以使用以下链接： triton-3.2.0+git8f9b005b-cp312-cp312-win_amd64.whl

安装完成后

运行 ComfyUI 并检查控制台日志。如果安装成功，日志中应显示类似以下内容（红框部分）：
[INFO] Triton and SageAttention successfully loaded.

常见问题及解决方法

1. 安装失败
   • 如果安装失败，请删除所有现有的 vs_buildtools、Windows SDK、vs_installer 和 CUDA，然后重新执行安装。
2. cuda_utils 加载失败
   • 如果在运行时遇到 DLL load failed while importing cuda_utils 错误，请尝试以下方法：
     • 删除 C:\Users\user_name\.triton 文件夹。
     • 参考 Issue #19 中的解决方案。

通过上述步骤，您可以轻松安装 Triton 和 SageAttention，从而在 ComfyUI 中实现更高效的图片和视频生成。如果遇到问题，请参考常见问题的解决方法或查阅相关文档。