PDM 避免安装 flash-attn 出现 No Module Named 'torch'

416 阅读1分钟

配置 CUDA_HOME,并且传入 no-isolation 参数,就可以正确安装了。

export CUDA_HOME=/usr/local/cuda-12.5
export PATH=$CUDA_HOME/bin:$PATH
pdm add flash-attn --no-isolation

参考来源