AMD显卡 ComfyUI 保姆级进阶手册

本教材编写于：2026.04.02，AI发展日新月异，后续可能不适配。

本教材将分步骤把特定版本 Triton和Sage-attention安装到 ComfyUI 环境中。

🟢 第一阶段：下载解压预编译安装包 (Wheel)

下载 Triton：
- 你会看到一个名为 Triton.Zip 的文件（大小约 220+ MB）。
解压文件：
- 将下载的 Triton.zip 解压。
- 里面会包含多个以 .whl 结尾的文件（这就是 Python 的安装包，也就是 Wheel 文件）。
- **将解压出来的文件复制到 C:\AI_Work)

🟡 第二阶段：在 ComfyUI 的独立环境中安装 Triton

🟢 针对本地环境的精确安装部署步骤

激活虚拟环境**。在 PowerShell 中输入：
Terminal window
```
cd C:\AI_Work
venv\Scripts\Activate
```
(执行成功后，你的命令行提示符最左侧应该会出现 (venv) 字样。)
安全安装 Triton Wheel 包

假设你之前下载并解压的 Triton wheel 文件放在了 Z:\artifact（路径替换为你实际路径）。

    # 根据自己的python版本选择对应的文件
    pip install "C:\AI_Work\Triton\triton_windows-3.6.0+gitae9d5a54.post27-cp312-cp312-win_amd64.whl" --no-deps --force-reinstall

--no-deps：这是关键，它告诉 pip “只安装 Triton，绝对不要动我现有的 PyTorch 和其他包”。
--force-reinstall：强制覆盖可能存在的旧版残留。

验证安装是否成功

安装完成后，紧接着在命令行输入以下命令，检查 Triton 是否已经正确挂载到你的 ROCm 环境下：

PowerShell
Terminal window
```
pip list | findstr "triton"
```
如果能正确输出 triton 的版本号，说明静态安装已经成功。

🔴 第三阶段：Sage-attention“部署

1.确认Sage-attention

安装 Sage-attention

PowerShell
Terminal window
```
python -m pip install sageattention==1.0.6 --no-deps --force-reinstall
```
*(💡 保姆级细节：加 ==1.0.6 是为了精准锁死在纯 Triton 架构的最后一个版本，阻断它去拉取带 CUDA C++ 的 v2 版本；加 --no-deps 是我们之前学到的保命神技，防止它破坏你的 Nightly 版 PyTorch。)
验证 SageAttention 版本
Terminal window
```
python -m pip list | findstr "sageattention"
```
- 如果输出是 sageattention 1.0.6：恭喜你，你已经处于 AMD 显卡在该领域的兼容性天花板。

2.替换加速文件

进入你的 SageAttention 安装目录：

C:\AI_Work\venv\Lib\site-packages\sageattention
把下载好的4个写满 Triton 算子代码的“战神版” Python 文件放进去替换掉：
- __init__.py
- attn_qk_int8_per_block.py
- attn_qk_int8_per_block_causal.py
- quant_per_block.py

🔴 第三阶段：测试

运行测试：

找一个工作流添
加 SAGE注意力补丁KJ节点，模式选择sageattn_qk_int8_pv_fp16_triton。
运行。

以下是修复错误的一个方法

🔴 如果开启Sage加速，出现类似错误：

!!! Exception during processing !!! Command '['C:\\Users\\wjluo\\AppData\\Local\\AMD\\ai_bundle\\ComfyUI\\venv\\Lib\\site-packages\\triton\\runtime\\tcc\\tcc.exe', 'C:\\Users\\wjluo\\AppData\\Local\\Temp\\tmps6ptnvus\\hip_utils.c', '-O3', '-shared', '-Wno-psabi', '-o', 'C:\\Users\\wjluo\\AppData\\Local\\Temp\\tmps6ptnvus\\hip_utils.cp312-win_amd64.pyd', '-fPIC', '-D_Py_USE_GCC_BUILTIN_ATOMICS', '-lpython312', '-LC:\\Users\\wjluo\\AppData\\Local\\AMD\\ai_bundle\\ComfyUI\\venv\\libs', '-IC:\\Users\\wjluo\\AppData\\Local\\AMD\\AI_Bundle\\ComfyUI\\venv\\Lib\\site-packages\\triton\\backends\\amd\\include', '-IC:\\Users\\wjluo\\AppData\\Local\\AMD\\AI_Bundle\\ComfyUI\\venv\\Lib\\site-packages\\_rocm_sdk_devel\\include', '-IC:\\Users\\wjluo\\AppData\\Local\\Temp\\tmps6ptnvus', '-IC:\\Users\\wjluo\\AppData\\Local\\AMD\\ai_bundle\\ComfyUI\\venv\\Scripts\\Include']' returned non-zero exit status 1.

1.检查自己的 \ComfyUI\venv 文件夹 2.如果有 Include 文件夹，里面有很多文件，比如：Python.h 3.把整个Include 文件夹复制到 \ComfyUI\venv\Scripts 4.彻底关闭 ComfyUI 并重新启动！

AMD显卡 ComfyUI 保姆级进阶手册-04