小白1分钟安装flash-attn

  1. 先在终端查看自己的python、torch、cuda的版本

  2. 在网站上查找对应的文件,下载下来:

Releases · Dao-AILab/flash-attention · GitHub

比如:我是python3.12、torch2.4、cuda 12.1、False

所以我下载的是这个文件:

【python 3.12,下载第三个文件,文件名:cp312对应的python3.12】

  1. 把这个文件传到你的文件夹里,然后直接

pip install flash_attn-2.8.3.post1+cu12torch2.4cxx11abiFALSE-cp312-cp312-linux_x86_64.whl.zip【也就是pip install 文件名】就ok啦!

相关推荐
2501_947575806 小时前
计算机毕业设计之jsp开山车行二手车交易系统
java·开发语言·hadoop·python·信息可视化·django·课程设计
骑士雄师6 小时前
java面试题 4:鉴权
java·开发语言
Byron__7 小时前
AI学习_06_短期记忆与长期记忆
人工智能·python·学习
时间的拾荒人7 小时前
C语言字符函数与字符串函数完全指南
c语言·开发语言
2501_948106917 小时前
计算机毕业设计之基于jsp教科研信息共享系统
java·开发语言·信息可视化·spark·课程设计
取经蜗牛8 小时前
Python 第一阶段完全指南:从零到第一个实用工具
开发语言·python
创世宇图8 小时前
【Python工程化实战】OpenTelemetry 在 Python 中的全链路追踪落地:从埋点到可视化的完整实战指南
python·分布式链路追踪·性能监控·opentelemetry·微服务可观测性
dog2508 小时前
从重尾到截断流量模型的演进
开发语言·php
qq_401700418 小时前
Qt QSS 完全入门写出漂亮界面以及解决样式不生效问题
开发语言·qt
许彰午9 小时前
72_Python爬虫基础BeautifulSoup
爬虫·python·beautifulsoup