小白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啦!

相关推荐
JustNow_Man1 小时前
psmux快捷键
人工智能·python
默子昂2 小时前
ollama 自定义ui
开发语言·python·ui
abcy0712132 小时前
Python中使用FastAPI和HDFS进行异步文件上传
python·fastapi
abcy0712132 小时前
flask hdfs 异步上传图文教程csdn
python·flask
在放️2 小时前
Python 爬虫 · PyQuery 模块基础
爬虫·python
装不满的克莱因瓶2 小时前
【自动驾驶领域】学习 Cityscapes 数据集——城市街景语义理解的标准基准
人工智能·pytorch·python·深度学习·学习·机器学习·自动驾驶
吴卫斌2 小时前
波动率控制仓位系列(一):满仓轮动的“过山车”困境
大数据·python·股票·量化交易
如此这般英俊2 小时前
手搓Claude Code-第三章 permission
人工智能·python·语言模型
TE-茶叶蛋3 小时前
TF-IDF 与 BM25 深度解析:从理论到项目实战
python·django·tf-idf