【bugfix】安装 flash-attn 报错

目录

1. 报错信息

2. 解决方法

---

安装 flash attention 报错

1. 报错信息

Building wheel for flash-attn (setup.py) ... error error: subprocess-exited-with-error ×

或者

Building wheel for flash-attn (pyproject.toml) did not run successfully 

甚至更多问题。

2. 解决方法

手动下载 wheel

https://github.com/Dao-AILab/flash-attention/releases/download/v2.3.3/flash_attn-2.3.3+cu122torch2.1cxx11abiFALSE-cp38-cp38-linux_x86_64.whl

然后

pip install flash_attn-2.3.3+cu122torch2.1cxx11abiFALSE-cp38-cp38-linux_x86_64.whl