qwen大模型，推理速度慢，单卡/双卡速度慢，flash-attention安装，解决方案

场景

阿里的通义千问qwen大模型，推理速度慢，单卡/双卡速度慢。
详细：
1、今日在使用qwen-14b的float16版本进行推理（BF16/FP16)
1.1 在qwen-14b-int4也会有同样的现象
2、使用3090 24G显卡两张
3、模型加载的device是auto，device=“auto”

解决方案

使用多卡推理，需要开启flash-attention，否则会慢

flash-attention安装

0、如果已经下载了qwen的源码，可以看到源码包里有flash-attention的文件夹。或者也可以去达摩院的git上下载：flash-attention的git地址

1、cd flash-attention
2、python setup.py install
2.1、在执行这句命令时，可能会报Could not build wheels for flash-attn, which is required to install pyproject.toml-based projects（如果不报，当我没说）

这里我的解决方法是执行

pip install flash-attn --no-build-isolation
1

还没结束，继续往下
3、至此就有了flash-attn包了，但是加载模型的时候，还是会报警告，这时的推理速度依旧是很慢的

Try importing flash-attention for faster inference...
Warning: import flash_attn rotary fail, please install FlashAttention rotary to get higher efficiency https://github.com/Dao-AILab/flash-attention/tree/main/csrc/rotary
Warning: import flash_attn rms_norm fail, please install FlashAttention layer_norm to get higher efficiency https://github.com/Dao-AILab/flash-attention/tree/main/csrc/layer_norm
1
2
3

4、继续安装rotary和layer_norm

# 安装rotary
cd flash-attention
cd csrc/rotary
python setup.py install

# 安装layer_norm
cd flash-attention
cd csrc/layer_norm
python setup.py install
1
2
3
4
5
6
7
8
9

5、至此安装完成，加载模型，不会报flash-attention的警告，加载速度也有显著的提升。

6、安装前，我尝试2048字数结果的问答，

• qwen-14b回答需要100秒，安装后需要70秒

• qwen-14b-int4回答需要60秒，安装后需要20秒