后端岗位面试题更新 2026-08-05
PyTorch 支持哪些模型并行化方法？在数据并行和模型并行时，哪些模块消耗显存？
寒武纪后端开发人工智能系统设计技术原理PyTorch
考察说明
考察分布式并行训练方法分类及显存消耗来源的理解
回答思路
- 能列举数据并行、模型并行、流水线并行、混合并行等主要方法
- 能区分数据并行与模型并行在显存占用上的差异
- 能指出参数、梯度、优化器状态、激活值等显存消耗模块
- 能说明同步与异步更新的影响，体现对边界条件的考虑
本题附完整参考答案与评分标准
登录后可查看结构化答题指导；也可以直接开一场模拟面试，AI 面试官用本题实时追问并给出评分。
