nvcc编译器安装

conda install -c nvidia cuda-nvcc==11.8.89

即可。

查看全部版本：

conda search -c nvidia cuda-nvcc

cuda toolkit安装：

conda install -c nvidia cuda-toolkit==11.8.0

装好后就可以编译flash attn了