Prhub

#43115 [CPU][DOC] Fix installation commands for Arm CPUs

原始 PR 作者 fadara01 合并时间 2026-05-20 07:31 文件变更 1 提交数 1 评论 2 代码增减 +5 / -5

执行摘要

修复 Arm CPU 安装命令以适配新 PyTorch

从 PyTorch 2.10 开始,PyPI 为 AArch64 Linux 提供 CUDA(而非仅 CPU)wheel,导致 Arm CPU 上的安装失败,错误提示找不到 torch==2.11.0+cpu。用户需要显式指定 CPU 索引才能正确安装。

建议合入,文档修复必要且及时。

讨论亮点

无 review 讨论。

实现拆解

  1. 修改安装指南:docs/getting_started/installation/cpu.arm.inc.md 中,对所有 uv pip install 命令追加 --torch-backend cpu 标志,对所有 pip install 命令追加 --extra-index-url https://download.pytorch.org/whl/cpu 标志,确保 pip/uv 从正确的源解析 PyTorch CPU wheel。
  2. 覆盖所有场景: 包括预编译 wheel 安装、nightly wheel 安装、指定版本安装等共 4 处命令,确保更新统一。
  3. 测试结果: 修改后安装命令可正常工作。
文件 模块 状态 重要度
docs/getting_started/installation/cpu.arm.inc.md 文档 modified 2.57

关键源码片段

docs/getting_started/installation/cpu.arm.inc.md documentation

唯一变更文件,修正 Arm CPU 安装命令

# 安装预编译 wheel(uv)
uv pip install https://github.com/vllm-project/vllm/releases/download/v${VLLM_VERSION}/vllm-${VLLM_VERSION}+cpu-cp38-abi3-manylinux_2_35_aarch64.whl --torch-backend cpu
​
# 安装预编译 wheel(pip)
pip install https://github.com/vllm-project/vllm/releases/download/v${VLLM_VERSION}/vllm-${VLLM_VERSION}+cpu-cp38-abi3-manylinux_2_35_aarch64.whl --extra-index-url https://download.pytorch.org/whl/cpu
​
# 从 nightly 索引安装(uv)
uv pip install vllm --extra-index-url https://wheels.vllm.ai/nightly/cpu --index-strategy first-index --torch-backend cpu
​
# 从 nightly 索引安装(pip,需指定完整 wheel 链接)
pip install https://wheels.vllm.ai/4fa7ce46.../vllm-0.13.0rc2...whl --extra-index-url https://download.pytorch.org/whl/cpu

添加 --torch-backend cpu--extra-index-url 确保从正确的 PyTorch CPU wheel 仓库拉取依赖。

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

风险极低。仅修改文档中的安装命令示例,不涉及任何代码逻辑。但需注意如果未来 PyTorch 行为再次变化,命令可能需同步更新。

直接影响 Arm CPU 用户的安装体验,解决因 PyTorch 索引变更导致的安装失败问题。对 x86 用户无影响。

文档变更无代码风险

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论