Prhub

#2172 [docker] Update training side dependencies

原始 PR 作者 zhuzilin 合并时间 2026-07-03 15:07 文件变更 2 提交数 5 评论 0 代码增减 +2 / -5

执行摘要

更新 Docker 训练依赖,升级 FlashLinearAttention 版本

保持训练镜像中依赖的最新兼容性,修复可能存在的 PyTorch 版本兼容问题(参考 comment 中关于 #168167 的说明)。

建议快速合并,是常规的依赖维护更新。

讨论亮点

无 review 评论。

实现拆解

  1. 升级 flash-linear-attention:在 docker/Dockerfile 中将 flash-linear-attention 版本从 0.4.1 升级到 0.4.2
  2. 移除 nvidia-cudnn-cu12 显式安装:删除 pip install nvidia-cudnn-cu12==9.16.0.29 行,因为该依赖已不再是必须或由基础镜像提供(关联 PyTorch issue #168167 已解决)。
  3. 更新版本号:在 docker/version.txt 中将版本从 nightly-dev-20260703a 更新为 nightly-dev-20260703b
  4. 提交历史包含 2 次 bugfix 调整,最终版本确定。
文件 模块 状态 重要度
docker/Dockerfile 部署脚本 modified 3.02
docker/version.txt 部署脚本 modified 1.32

分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。

评论区精华

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

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

风险与影响

低风险。变更仅限于依赖版本升级和移除一个可选依赖,不涉及核心逻辑。但 flash-linear-attention 升级可能引入微小的行为变化,需确保与当前模型训练兼容。

影响范围仅限于 Docker 训练镜像的用户。升级 flash-linear-attention 可能带来性能或功能改进,移除 nvidia-cudnn-cu12 可减少镜像大小并避免潜在冲突。

依赖版本变更

关联 Issue

未识别关联 Issue

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

完整报告

参与讨论