ndidia显卡:cuda计算能力 —— cuda sm_xx —— flash_attn nvcc fatal : Unsupported gpu architecture 'compute_120'

安装flash-attention报错:


flash_attn nvcc fatal : Unsupported gpu architecture 'compute_120'




image




总结原因:

  1. 显卡显卡不支持,估计显卡过老,最好更换5060ti以上显卡;

  2. cuda版本不匹配;




参考:

https://developer.baidu.com/article/details/3301013


不支持的GPU架构: ‘nvcc fatal: Unsupported gpu architecture ‘compute_XX’’错误通常意味着你正在尝试编译或运行的代码不支持你的GPU架构。NVIDIA的CUDA工具链(包括nvcc编译器)针对不同的GPU架构有不同的’compute’版本,例如’compute_30’、’compute_35’等。如果你的代码是为特定版本的GPU架构编译的,而你的GPU架构与之不匹配,就会出现这个错误。


CUDA版本不匹配: ‘RuntimeError: Error compiling objects for extension’错误可能是由于PyTorch的CUDA版本与你的GPU驱动或CUDA工具链版本不匹配造成的。如果你的PyTorch是在CUDA 10.1上编译的,但你的系统中安装的是CUDA 11或更高版本,就可能出现这个错误。




posted on 2026-04-03 15:03  Angry_Panda  阅读(39)  评论(0)    收藏  举报

导航