【环境部署】安装flash-attention

bash 复制代码
FlashAttention:具有 IO 感知能力的快速且内存高效的精确注意力机制
https://github.com/Dao-AILab/flash-attention

安装方式

pip 安装失败,需要官方下载whl,离线安装,下载地址

bash 复制代码
# linux版本
https://github.com/Dao-AILab/flash-attention/releases?page=1
# Windows版本
https://github.com/kingbri1/flash-attention/releases
  • cu12torch2.4->代表安装了torch2.4+cu12的版本
  • cp310->代表python版本
相关推荐
ken223212 小时前
怎样选择 python 包管理器 pip, uv, conda ?
pip·uv
万粉变现经纪人2 天前
如何解决 pip install tensorflow-gpu 报错 未检测到 CUDA 驱动 问题
人工智能·python·深度学习·aigc·tensorflow·bug·pip
baidu_huihui2 天前
在 CentOS 9 上安装 pip(Python 的包管理工具)
开发语言·python·pip
万粉变现经纪人2 天前
如何解决 import aiohttp ModuleNotFoundError: No module named ‘aiohttp’
python·scrapy·beautifulsoup·aigc·pillow·pip·httpx
万粉变现经纪人3 天前
如何解决 pip install ta-lib 报错 本地 TA-Lib 库未安装 问题
数据库·python·scrapy·oracle·bug·pandas·pip
一勺汤3 天前
YOLO26 改进、魔改| 空间-通道特征调制器SCFM,通过空间与通道双维度注意力调制,融合全局与局部特征,提升视觉任务的特征质量与模型效果。
yolo·注意力机制·降噪·小目标·yolo26·yolo26改进
tryCbest5 天前
Pip生成requirements.txt文件
python·pip
李昊哲小课7 天前
pip缓存配置
python·缓存·pip
做cv的小昊8 天前
【conda】打包已有conda环境并在其他服务器上搭建
运维·服务器·python·conda·运维开发·pip·开发
时光之源9 天前
程序猿常用命令行(Linux、Windows、Powershell、CMD、conda、pip、apt)
linux·conda·pip