检查环境
检查环境
python -c "import flash_attn, torch; print(flash_attn.__version__, torch.__version__)"
# 推理框架通常自带匹配编译产物:优先用框架 wheel,而不是单独 pip 装 flash-attn| 收益 | 注意力部分速度与显存双赢 |
|---|---|
| 约束 | 架构与版本矩阵较严(Ampere 及更新为主) |
| 实践 | 通过 vLLM/TGI 等镜像获取,减少手工编译 |
报错关键词如 "flash_attn" 加 "undefined symbol" 基本是版本错配,按框架要求的组合重装。