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

如果返回下面相关的代码则正面triton成功安装

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.1.1+cu128torch2.7.0-cp312-cp312-win_amd64.whl
输入以下代码安装:

python.exe -m pip install H:\Software\Xformers\sageattention-2.1.1+cu128torch2.7.0-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

Image
Cuda12.8环境下升级Comfyui的triton于SageAttention-数字折叠
Cuda12.8环境下升级Comfyui的triton于SageAttention
此内容为付费资源,请付费后查看
会员专属资源
您暂无购买权限,请先开通会员
开通会员
提供远程协助
支持需求定制
适用软件ComfyUI
适配环境cu12.8py312
大小3.25GB
工具使用问题请联系微信 15915765126
付费资源
© 版权声明
THE END
喜欢就支持一下吧
点赞15 分享
评论 抢沙发

请登录后发表评论

    暂无评论内容