Forum

Notifications
Clear all

Flash Attention 对本地推理有帮助吗?我需要手动开吗?

0 帖子
1 Users
0 Reactions
9 查看
0
[#60]
Topic starter

【官方常见问题精选】欢迎补充和追问。

Flash Attention 是一类更省显存、往往更快的注意力实现,在训练与部分推理框架中常见。对终端用户而言,通常已封装在框架/内核里,而不是在聊天 UI 里点一个按钮。

建议:

  • 用 vLLM、较新的 PyTorch/CUDA 栈或官方容器时,优先跟随文档的推荐环境。
  • GGUF + llama.cpp 路线有自己的内核优化,不必强行套用「在 Python 里装 flash-attn」的经验。
  • 版本不匹配时编译失败很常见,先保证能稳定出字,再追求极致。

欢迎补充某一框架下启用前后的显存/速度对比(注明版本)。


Share: