训练YOLO12的时候报错:
在这里插入图片描述
本机环境:
CUDA11.8 + torch2.5 + py3.10 + flash-attn==2.7.3

解决:
首先确认一遍 flash-attn的版本与本机环境是否对应。
如果对应还是出现这个问题,可以换个版本,如2.7.0.post1

flash-attn下载地址:

https://github.com/Dao-AILab/flash-attention/releases

在这里插入图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐