配置 CUDA_HOME,并且传入 no-isolation 参数,就可以正确安装了。
export CUDA_HOME=/usr/local/cuda-12.5
export PATH=$CUDA_HOME/bin:$PATH
pdm add flash-attn --no-isolation
配置 CUDA_HOME,并且传入 no-isolation 参数,就可以正确安装了。
export CUDA_HOME=/usr/local/cuda-12.5
export PATH=$CUDA_HOME/bin:$PATH
pdm add flash-attn --no-isolation