YOLO12提示:FlashAttention is not available on this device. Using scaled_dot_product_attention instead
·
训练YOLO12的时候报错:
本机环境:
CUDA11.8 + torch2.5 + py3.10 + flash-attn==2.7.3
解决:
首先确认一遍 flash-attn的版本与本机环境是否对应。
如果对应还是出现这个问题,可以换个版本,如2.7.0.post1
flash-attn下载地址:

更多推荐
所有评论(0)