拿下技术面上传简历

集群是 8 个节点、每节点 8 卡,节点内 NVLink 高带宽,节点间只有 25Gbps 以太网。现在要训一个 70B 模型。请说明你会怎么组合数据并行、张量并行、流水并行(各维并行度大概取多少),并且解释每一维并行为什么必须/可以放在对应的互联层级上。

深度学习基础训练工程1-3 年知识点 1 个

面试官在考什么

考察是否理解三种并行各自的通信模式与通信量,从而能做拓扑感知的并行策略设计,而不是只会调并行度参数。

数据并行/模型并行

回答框架 · 前两步

  1. 张量并行把单层的权重矩阵按行/列切开,每层前向/反向都要做 all-reduce 或 all-gather,通信频率高、量大,必须压在高带宽的 NVLink 域内,通常 TP ≤ 8(即不跨节点)。
  2. 流水并行把模型按层切成若干 stage,只在 stage 边界传 activation,通信量小、频率低,可以跨节点;代价是 bubble 和调度复杂度。

还有 3 条答案要点、 3 层追问、 5 个常见坑, 以及针对你简历的追问路径,在定制报告里。

上传简历,生成我的题库先看示例报告

延伸阅读

想看这个方向的整体准备路线? 进入「深度学习基础」方向页

同一能力域的其他题