首先按照Meshanything官方配备环境后,执行命令
pip install flash-attn --no-build-isolation
时build wheel 超过10小时未成功。进行检查环境:
nvcc -V
CUDA 12.1
pytorch 2.3.1
编译时产生错误
ImportError: DLL load failed while importing flash_attn_2_cuda: 找不到指定的程序
利用FlashAttention源码clone后编译
python setup.py install
编译种……