Windows下安装SageAttention
自己编译
需要先安装好python环境和对应的pytorch,最好另外建一个虚拟环境,虽然直接在本地上也可以编译。
git clone https://github.com/thu-ml/SageAttention.git
cd SageAttention
python setup.py bdist_wheel
编译完成后在dist文件夹可以找到
sageattention-2.2.0-win_amd64.whl
如果在本机环境编译并安装
python setup.py install # or pip install -e .
编译环境参考,请勿直接照搬
直接安装别人编译好的whl
需要注意对应python版本,torch版本,系统版本。
例如sageattention-2.2.0+cu128torch2.7.0-cp312-cp312-win_amd64.whl
cu128代表CUDA12.8,torch2.7.0代表pytorchtorch2.7.0,cp312代表python3.12,win_amd64代表64位Windows环境。
下面搜集了一些大佬编译好的版本,需要魔法访问
Releases · sdbds/SageAttention-for-windows
Kijai/PrecompiledWheels at main
Releases · woct0rdho/SageAttention