执行摘要

• 一句话：修复 aarch64/GB200 稳定 SGLang 镜像构建
• 推荐动作：建议合并后尽快创建后续 PR 修复 review 中提出的问题：恢复 transformers==5.3.0 的条件安装，并将 pip show 替换为安全的 echo 或 : 命令。此外，建议在 CI 中增加对 aarch64 和 x86_64 双架构的构建验证。

功能与动机

原有的 Dockerfile 在 aarch64（GB200）上构建失败，因为 flash-attn-4 和 sglang-kernel 的 x86_64 平台 Wheel 包不兼容 arm 架构。PR body 提到目标是在 GB200 上实现稳定构建。Review 中也指出 transformers==5.3.0 的安装被意外移除，影响了 x86_64 构建。

实现拆解

1. 架构检测与条件分支：在 docker/Dockerfile.stable.sglang 中，将原本无条件卸载并重新安装 flash-attn-4 和 sglang_kernel 的命令改为基于 $(uname -m) 的条件判断：
   • 对于 aarch64 架构，使用 pip show 保留 SGLang 基础镜像中自带的 arm 版本，不再重新安装。
   • 对于其他架构（x86_64），保持原有行为：卸载 flash-attn-4 后安装 flash-attn-4[cu13]，并通过 wget 下载并强制重新安装指定版本的 sglang_kernel。
2. 依赖移除处理：transformers==5.3.0 的安装行被整体删除，但 review 指出这个删除是无意的，且会影响 x86_64 构建。不过当前 PR 未修复此问题。
3. 测试验证：作者在 GB200 节点上手动验证了构建成功，并确认 sglang-kernel 包含 SM100 核心、CUDA matmul/Apex 烟雾测试/ SGLang CLI 均通过；x86_64 未做验证。

关键文件：

• docker/Dockerfile.stable.sglang（模块 Docker；类别 infra；类型 infrastructure）: 唯一变更文件，包含所有架构条件分支逻辑，直接影响 SGLang 稳定镜像构建。

关键符号：未识别

关键源码片段

docker/Dockerfile.stable.sglang

唯一变更文件，包含所有架构条件分支逻辑，直接影响 SGLang 稳定镜像构建。

# 根据架构选择是否重新安装 flash-attn-4
# aarch64 使用基础镜像自带的版本，x86_64 卸载后安装带 CUDA 加速的版本
RUN if [ "$(uname -m)" = "aarch64" ]; then \
    # 只在 aarch64 上检查包是否存在，未安装时 pip show 会返回非零退出码
    pip show "flash-attn-4"; \
else \
    pip uninstall -y "flash-attn-4" && \
    pip install "flash-attn-4[cu13]"; \
fi
​
# 同理处理 sglang-kernel
RUN if [ "$(uname -m)" = "aarch64" ]; then \
    pip show "sglang-kernel"; \
else \
    wget https://github.com/sgl-project/whl/releases/download/v0.4.2.post2/sglang_kernel-0.4.2.post2+cu130-cp310-abi3-manylinux2014_x86_64.whl#sha256=4e7ce619274234d182b20da883fcf1d20e7e55cbea90d62244e2b6a3d6c0fc85 && \
    pip install sglang_kernel-0.4.2.post2+cu130-cp310-abi3-manylinux2014_x86_64.whl --force-reinstall; \
fi

评论区精华

主要的 review 评论来自 gemini-code-assist[bot]，指出了两个问题：

1. transformers==5.3.0 被意外移除：该行被完全删除，影响 x86_64 构建，建议也使用条件安装。
2. pip show 作为空操作可能导致构建失败：建议替换为 echo 以确保安全。
   此外，wuxibin89 已经批准了 PR。这些评论在 PR 合并前未得到回复或修复。

• transformers 版本被意外移除 (correctness): 未修复，PR 已合并但 transformers 问题遗留。
• pip show 作为空操作不安全 (correctness): 未修复，PR 已合并但该问题遗留。

风险与影响

• 风险：
  1. x86_64 构建退化：transformers==5.3.0 被意外移除，可能导致 x86_64 构建使用错误的 transformers 版本，引发兼容性问题或构建失败。
  2. aarch64 构建隐患：使用 pip show 作为空操作，如果基础镜像中没有预装这些包，pip show 会返回非零退出码，导致镜像构建失败。
  3. 版本锁定缺失：aarch64 路径下未锁定 flash-attn-4 和 sglang_kernel 版本，依赖于基础镜像中的版本，未来更新可能导致不确定的行为。
• 影响：
  • 影响用户：GB200（aarch64）用户可以成功构建 SGLang 稳定镜像；x86_64 用户可能遇到 transformers 相关问题。
  • 影响系统：Dockerfile 的变更将影响所有使用 Dockerfile.stable.sglang 的 CI 流程和本地构建。
  • 影响团队：需要关注 x86_64 构建 CI 是否通过，并决定是否在后续 PR 中修复 transformers 和 pip show 问题。
  • 风险标记：缺少 x86_64 测试验证, transformers 版本移除, pip show 空操作风险

关联脉络

• PR #6573 [doc] chore: Update ascend_sglang_best_practices.rst: 都涉及 SGLang 在 NPU/Ascend 平台的使用，本 PR 提供构建支持。
• PR #6532 [doc] refactor: optimize ascend doc: 都涉及 Ascend 平台基础设施，本 PR 提供 Docker 构建支持。