最近安装vllm的时候 有时候pytorch版本总是弄错,这里写下vllm-flash-attn与pytoch对应关系

打开网站

vllm-flash-attn · PyPI

查询历史版本

点进去查询对应的pytoch

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐