一起来加速!SageAttention + 5060Ti 助力AI绘画效率飞升2.7倍指南

360影视 动漫周边 2025-05-06 00:36 2

摘要:NVIDIA RTX 5060Ti 16G,可能是目前AI绘画爱好者最值得入手的一张显卡了——价格适中,显存够大。目前,ComfyUI 也及时提供了最新版适用于50系显卡的便携包。

NVIDIA RTX 5060Ti 16G,可能是目前AI绘画爱好者最值得入手的一张显卡了——价格适中,显存够大。目前,ComfyUI 也及时提供了最新版适用于50系显卡的便携包。

新安装 NVIDIA RTX 5060Ti 16G之后,使用 ComfyUI 最新便携版(Python 3.13.2 + Pytorch 2.80+cu128),安装 SageAttention 可以使绘画效率大幅提升。具体方法如下:

一、推荐使用 fp8大模型,可加速50系显卡生图。

二、安装使用 Triton 与 sage attention 可加速50系显卡运算。

1、下载 triton 编译文件(triton_windows-3.3.0a0.post18-cp313-cp313-win_amd64.whl);

2、下载 sage attention 编译文件(sageattention-2.1.1+cu128torch2.8.0-cp313-cp313-win_amd64.whl);

地址如下:

https://github.com/woct0rdho/SageAttention/releases/download/v2.1.1-windows/sageattention-2.1.1+cu128torch2.8.0-cp313-cp313-win_amd64.whl

3、下载 ComfyUI 便携版 缺失的 python 文件夹(libs 和 include);

地址如下:

4、在ComfyUI目录中执行CMD安装命令(文件地址请自行更替),比如:

python_embeded\python.exe -m pip install H:\comfyui_windows\triton_windows-3.3.0a0.post18-cp313-cp313-win_amd64.whl

python_embeded\python.exe -m pip install H:\comfyui_windows\sageattention-2.1.1+cu128torch2.8.0-cp313-cp313-win_amd64.whl

5、将 libs 和 include 文件夹复制到 python_embeded 文件夹中。

6、编辑 ComfyUI 启动文件 run_nvidia_gpu_fast.bat,添加 --use-sage-attention 启动提示代码,保存即可。

安装成功后,系统会提示 Using sage attention

三、实际效果,出图效率提高2.7倍!

刚开始安装 5060Ti 之时,使用 Flux(GGUF)大模型出一张图(768*1280)大约在120秒左右;

而使用 Flux(FP8)大模型 + sage attention 之后,出一张图(768*1280)大约在44秒左右(2.48s/it);

提升了2.7倍。

而我们通常出图之后还会使用扩图修复节点,出大图(960*1600)的效率也从180秒提升到了71秒左右(3.88s/it);

提升了2.5倍。

所以,整个工作流的出图效率从300秒/张,提升到了115秒/张,提升2.6倍(画质不变)。

这与 sage attention 官方宣称的提升效率基本一致。

注:triton 是 sage attention 的依赖项,请先安装 triton;

注:ComfyUI 缺失的 libs 与 include 文件夹,似乎并不影响 sage attention 的正常运行(已验证)——这是由于 sage attention 的编译者最近改进了安装文件。但 ComfyUI 启动时,依然会发出警告提示缺失这两个文件夹,所以,推荐补全这两个文件夹。

来源:叶培盛一点号

相关推荐