如何在pixi.toml中配置flash_attn并禁用构建隔离

问题根源 #

你手动执行pixi run pip install flash_attn --no-build-isolation能成功，是因为该命令跳过了构建隔离，直接使用当前环境中的torch完成构建。但自行添加配置时，要么未在依赖中声明torch，要么extra-build-dependencies中的包名不匹配（PyPI包名为flash-attn而非flash_attn），导致构建时找不到torch。

正确配置方案 #

方案一：沿用--no-build-isolation逻辑（推荐，与手动命令一致）

将以下配置写入pixi.toml：

[pypi-dependencies]
# 必须添加torch，flash_attn的运行和构建都依赖它
torch = "*"
flash_attn = "*"

[tool.pixi.pypi-options]
# 对flash_attn禁用构建隔离，使用环境中的torch构建
no-build-isolation = ["flash_attn"]

执行命令完成安装：

pixi install

方案二：使用构建隔离+额外构建依赖

若不想禁用构建隔离，可修正包名并添加构建依赖：

[pypi-dependencies]
torch = "*"
flash_attn = "*"

[tool.uv.extra-build-dependencies]
# 注意包名是flash-attn（连字符），与PyPI上的包名一致
flash-attn = ["torch"]

同样执行pixi install完成安装。

为什么之前的配置失败？ #

1. 未在[pypi-dependencies]中声明torch：启用no-build-isolation时，构建会依赖环境已安装的包，但pixi未安装torch，导致构建报错。
2. extra-build-dependencies中的包名错误：PyPI上的包名是flash-attn，而非flash_attn，uv无法匹配到对应包，导致构建环境中未安装torch。

内容的提问来源于stack exchange，提问作者j35t3r