Prhub

#26267 [NPU] Add torchaudio dependency for NPU platform

原始 PR 作者 xdtbynd 合并时间 2026-05-25 16:30 文件变更 2 提交数 2 评论 1 代码增减 +3 / -1

执行摘要

NPU 平台依赖补齐 torchaudio

NPU 平台需要 torchaudio 作为依赖以支持音频相关模型处理。当前 NPU Docker 镜像和 pyproject 配置缺少该包,在使用音频功能时会导致运行时 import 错误。

建议阅读,NPU 平台维护者可了解依赖补齐的标准模式。不过,建议后续补充 check_env.py 的更新以完善诊断能力。

讨论亮点

review 中 gemini-code-assist[bot] 建议同时更新 python/sglang/check_env.py,将 torchaudio 加入 PACKAGE_LIST 或 NPUEnv.EXTRA_PACKAGE_LIST,以便环境诊断时能正确报告版本,辅助调试 NPU 上的音频问题。但该建议未被采纳,PR 已直接合并。

实现拆解

  1. 修改 Dockerfile (docker/npu.Dockerfile):新增 ARG TORCHAUDIO_VERSION="2.8.0" 并在 PyTorch 安装命令中追加 torchaudio==${TORCHAUDIO_VERSION},确保 NPU 镜像构建时安装该依赖。
  2. 修改项目依赖文件 (python/pyproject_npu.toml):在 dependencies 列表中插入 "torchaudio==2.8.0",使通过 pip 安装 NPU 依赖时自动引入 torchaudio。
文件 模块 状态 重要度
docker/npu.Dockerfile 部署脚本 modified 2.88
python/pyproject_npu.toml 项目配置 modified 2.93

关键源码片段

docker/npu.Dockerfile infrastructure

NPU Docker 镜像构建入口,新增 TORCHAUDIO_VERSION 参数并修改 PyTorch 安装命令,确保镜像中包含 torchaudio。

# docker/npu.Dockerfile (partial)
ARG TORCHAUDIO_VERSION="2.8.0" # 新增:定义 torchaudio 版本
​
# 修改后的 PyTorch 安装命令
RUN . /etc/environment_new && \
    (${PIP_INSTALL} torch==${PYTORCH_VERSION} torchvision==${TORCHVISION_VERSION} torchaudio==${TORCHAUDIO_VERSION} --index-url https://download.pytorch.org/whl/cpu) \
    && (${PIP_INSTALL} ${PTA_URL})
python/pyproject_npu.toml configuration

NPU 平台的 Python 依赖清单,新增 torchaudio==2.8.0 确保 pip 安装时自动拉取。

# python/pyproject_npu.toml (partial)
dependencies = [
    ...
    "torchao==0.9.0",
    "torchaudio==2.8.0", # 新增:音频处理依赖
    "tqdm",
    ...
]

评论区精华

建议更新 check_env.py 以包含 torchaudio 版本诊断 other

gemini-code-assist[bot] 在 pyproject_npu.toml 的 review 评论中建议:" 当为 NPU 平台添加 torchaudio 依赖时,建议同时更新 python/sglang/check_env.py,将 torchaudio 添加到 PACKAGE_LIST 或 NPUEnv.EXTRA_PACKAGE_LIST,以便环境诊断能正确报告版本,辅助调试 NPU 上的音频问题。"

结论:PR 作者未采纳该建议,PR 直接合并。该建议无害但非必需,依赖补全本身即可满足功能需求。 · 已解决

风险与影响

风险极低。仅新增依赖项,不涉及任何模型逻辑或核心代码变更。torchaudio 版本固定为 2.8.0,与 PyTorch 2.8.0 配合,版本兼容性风险小。未更新 check_env.py 可能导致环境诊断遗漏 torchaudio 版本信息,但不会影响运行。

影响范围窄,仅作用于 NPU 平台的 Docker 镜像和 pip 安装,确保音频功能可用。对其他平台无影响。

环境诊断遗漏

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论