小米集团面试题更新 2026-08-05

在LoRA训练的时候,用多少资源,用的什么卡,占用多少显存,一个模型有哪些模块是占用显存的,分别分析一下占用大小

小米集团人工智能电子/半导体性能优化系统设计技术原理LoRA

考察说明

考察对LoRA训练资源需求、GPU显存占用构成及模型各模块显存分布的深入理解

回答思路

  1. 结合模型规模、训练配置、批次大小等说明资源需求
  2. 明确所用GPU型号及其显存大小
  3. 逐一分析模型各模块(如参数、梯度、优化器状态、激活值、LoRA参数)的显存占用
  4. 给出估算方法及调优思路