Hey everyone! I’ve successfully set up Triton and Sage Attention on my RTX 5070Ti with Windows 11 to speed up Wan2.1 generation. This guide is for anyone who doesn’t want to mess with WSL or install Linux. Here are the steps, with references for more details:<a target="_blank" rel="ugc" href="https://github.com/woct0rdho/triton-windows/?tab=readme-ov-file">https://github.com/woct0rdho/triton-windows/?tab=readme-ov-file</a><a target="_blank" rel="ugc" href="https://www.reddit.com/r/comfyui/comments/1j298vc/comfyui_not_recognizing_sageattention/">https://www.reddit.com/r/comfyui/comments/1j298vc/comfyui_not_recognizing_sageattention/</a><a target="_blank" rel="ugc" href="https://github.com/thu-ml/SageAttention/issues/107">https://github.com/thu-ml/SageAttention/issues/107</a>1. Download the special ComfyUI version for 50XX series GPUs, extract it, and run the run_nvidia_gpu.bat batch file. Wait until all dependencies are installed.<a target="_blank" rel="ugc" href="https://github.com/comfyanonymous/ComfyUI/releases/download/latest/ComfyUI_cu128_50XX.7z">https://github.com/comfyanonymous/ComfyUI/releases/download/latest/ComfyUI_cu128_50XX.7z</a>2. Install MSVC and Windows SDK<ul><li>Steps:<ul><li>Install Visual Studio 2022 from <a target="_blank" rel="ugc" href="https://visualstudio.microsoft.com/downloads/">Visual Studio Downloads</a>. Select “Desktop development with C++”.</li><li>Add to PATH:<ul><li>Locate cl.exe (e.g., C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\VC\Tools\MSVC\14.43.34808\bin\Hostx64\x64).</li><li>Add to PATH:cmd<code>set PATH=%PATH%;C:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\VC\Tools\MSVC\14.43.34808\bin\Hostx64\x64</code></li><li>Or set permanently via System &gt; Advanced &gt; Environment Variables &gt; Edit “Path”.</li></ul></li><li>Test:<ul><li>OpenPowerShell and enter <code>cl</code></li><li>Expected: Microsoft (R) C/C++ Optimizing Compiler Version 19.43.34808 for x64....</li></ul></li></ul></li></ul>3. Install vcredist<ul><li>Steps:<ul><li>Download Visual C++ Redistributable (2015-2022) from <a target="_blank" rel="ugc" href="https://aka.ms/vs/17/release/vc_redist.x64.exe">Microsoft</a>.</li><li>Run vc_redist.x64.exe and install.</li></ul></li></ul>4. Install Triton<ul><li>Steps:<ul><li>Install a pre-release wheel for Windows:cmd<code>cd ComfyUI_cu128_50XX\python_embeded python.exe -m pip install https://github.com/woct0rdho/triton-windows/releases/download/v3.2.0%2Bgit8f9b005b-windows.post11/triton-3.2.0+git8f9b005b-cp312-cp312-win_amd64.whl</code></li></ul></li></ul>5. Install Python 3.12.8 and Configure ComfyUI Embedded Python<ul><li>Why: ComfyUI’s embedded Python needs headers and libs for Triton/Sage Attention.</li><li>Steps:<ul><li>Download Python 3.12.8 from <a target="_blank" rel="ugc" href="https://www.python.org/downloads/release/python-3128/">Python.org</a> (Windows installer).</li><li>Install to a temporary path (e.g., C:\Python312).</li><li>Copy include and libs to ComfyUI’s embedded Python:cmd<code>xcopy C:\Python312\include C:\Data\ComfyUI_cu128_50XX\python_embeded\include</code><code>xcopy C:\Python312\libs C:\Data\ComfyUI_cu128_50XX\python_embeded\libs</code></li></ul></li></ul>6. Install Sage Attention 1.0.6<ul><li>Steps:<ul><li>Download Sage Attention 1.0.6 <a target="_blank" rel="ugc" href="https://github.com/thu-ml/SageAttention/tree/sageattention-1">https://github.com/thu-ml/SageAttention/tree/sageattention-1</a></li><li>Extract it to the Comfy UI folder and replace <a target="_blank" rel="ugc" href="http://setup.py">setup.py</a> with the attached file</li><li>Install using embedded Python:cmd<code>cd C:\Temp\sageattention-1.0.6 \ComfyUI_cu128_50XX\python_embeded\python.exe setup.py install</code></li></ul></li></ul>Before and After Results<ul><li>Test Setup: 512x512 i2v video with 30 steps, 41 frames, 20 block swaps and UniPC sampler using I2V 720p 8fp model from master Kijai <a target="_blank" rel="ugc" href="https://huggingface.co/Kijai/WanVideo_comfy/tree/main">https://huggingface.co/Kijai/WanVideo_comfy/tree/main</a>.</li><li>SDPA Attention: ~27s/it.<img src="https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/fbe816a3-8d95-4936-83db-ad78488bad8e/width=525/fbe816a3-8d95-4936-83db-ad78488bad8e.jpeg" /></li><li>Sage Attention: ~14s/it.</li><li><img src="https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/b062d9fc-c21b-431d-b4d0-6ce8b7fee172/width=525/b062d9fc-c21b-431d-b4d0-6ce8b7fee172.jpeg" />Sage Attention+TeaCache: ~8s/it<img src="https://image.civitai.com/xG1nkqKTMzGDvpLrqFT7WA/1a6cd45b-baea-49fa-969c-56456808b1bb/width=525/1a6cd45b-baea-49fa-969c-56456808b1bb.jpeg" /></li></ul>

Files with SHA256: 9c6e70c0a200cc404f12a6d96b600205242e2adcd947947a12337fcde193bf3c