Cuda12.8环境下升级Comfyui的triton与SageAttention

问题描述

一些新的功能需要更高的cuda驱动版本，但是我们升级完cuda之后，之前的一些配套环境也需要先升级为相匹配的版本。

升级方法

PyTorch

1.安装相匹配的Pytorch轮子：download.pytorch.org/whl/torch/
我们提供的是torch-2.7.0+cu128-cp312-cp312-win_amd64.whl
如果之前有安装需要先通过以下代码卸载：

python.exe -m pip uninstall torch -y
python.exe -m pip uninstall torchvision torchaudio -y

然后安装下载的轮子（轮子路径为你下载的路径，下面所有代码同）

python.exe -m pip install H:\Software\Xformers\torch-2.7.0+cu128-cp312-cp312-win_amd64.whl

2.下载相匹配的torchvision：https://download.pytorch.org/whl/cu128/torchvision/
我们提供的是torchvision-0.22.1+cu128-cp312-cp312-win_amd64.whl
安装代码

python.exe -m pip install --no-deps H:\Software\Xformers\torchvision-0.22.1+cu128-cp312-cp312-win_amd64.whl

triton

1.先卸载原有的triton：

python.exe -m pip uninstall triton -y

2安装新的triton：

python.exe -m pip install -U "triton-windows<3.4"

3.将我们提供的python_3.12.7_include_libs文件夹里的所有文件夹复制到你的ComfyUI的环境根目录里,例如 H:\ComfyUI\python_embeded，然后在环境里通过以下代码验证triton是否安装成功

python.exe test.py

如果返回下面相关的代码则证明riton成功安装

tensor([0., 0., 0.], device='cuda:0')
If you see tensor([0., 0., 0.], device='cuda:0'), then it works

SageAttention

1.卸载原有的SageAttention:

python.exe -m pip uninstall sageattention -y

2.下载相匹配的sageattention ：Releases · woct0rdho/SageAttention
我们提供的是sageattention-2.2.0+cu128torch2.7.1-cp312-cp312-win_amd64.whl（支持torch2.7..0）
输入以下代码安装：

python.exe -m pip install H:\Software\Xformers\sageattention-2.2.0+cu128torch2.7.1-cp312-cp312-win_amd64.whl

Xformes

请查看以下链接：ComfyUI正确安装Xformes的方法-数字折叠

其他说明

在升级前请确保已经安装Visual Studio软件以及cuda驱动升级为12.8版本

cuda12.8下载链接：CUDA Toolkit 12.8 Downloads | NVIDIA Developer

参考视频：https://www.bilibili.com/video/BV1sDK3zzE27/?spm_id_from=333.337.search-card.all.click&vd_source=8640822e652481be20f735ab16dad292