Install Windows Triton And Sage Attention
Added 2025-04-01 20:35:26 +0000 UTCStep 1 - Modify your Visual Studio Build Tools 2022 (Or Install It)
Step 2 - Verify / Install MSVC v143 + Windows 11 SDK
Step 3 - Install CUDA Toolkit 12.5
https://developer.nvidia.com/cuda-12-5-0-download-archive
Add the CUDA Toolkit 12.5 path to your system environment variables (and remove any old ones, they will break this.)
Step 4 - Go to triton-windows GitHub and READ IT.
https://github.com/woct0rdho/triton-windows/
Step 5 - Add this to your path system environment variable
Step 6 - Download the .whl file that corresponds to your python version (check triton github for portable install specific instructions)
https://github.com/woct0rdho/triton-windows/ (Click on Releases)
(do not use Triton ver.2!!! )
In my case it was π
pip install triton-3.1.0-cp310-cp310-win_amd64.whl
pip install sageattention
Reference from : https://purz.notion.site/Get-Windows-Triton-working-for-Mochi-6a0c055e21c84cfba7f1dd628e624e97
But we are not only using this for Mochi. This can be use for other AI Video Models.
Comments
nice job. tyvm
Jonathan Bach
2025-04-02 21:09:02 +0000 UTCTysm, deeply appreciated!
Melting Province
2025-04-01 20:38:48 +0000 UTC