详解 ZeRO 冗余消除技术:为什么 ZeRO-3 能让单卡跑起“塞不下”的模型
为什么 ZeRO-3 能让单卡跑起“塞不下”的模型? 随着大语言模型(LLM)的尺寸不断膨胀,GPU的显存往往成为训练过程中的最大瓶颈。一个1750亿参数的模型(如GPT-3),即使使用混合精度(FP16/BF16),仅参数、梯度和优化器状...
为什么 ZeRO-3 能让单卡跑起“塞不下”的模型? 随着大语言模型(LLM)的尺寸不断膨胀,GPU的显存往往成为训练过程中的最大瓶颈。一个1750亿参数的模型(如GPT-3),即使使用混合精度(FP16/BF16),仅参数、梯度和优化器状...
在PyTorch的深度学习训练中,管理和清零梯度(Gradient)是一个基础且关键的步骤。然而,很多初学者在尝试手动清零某个特定张量(Tensor)的梯度时,会遇到一个令人困惑的错误:AttributeError: ‘Tens...
在训练或部署超大规模AI模型(如千亿参数LLM)时,GPU显存(VRAM)是最大的瓶颈。尽管单卡显存容量不断提升,但模型增长速度更快。解决这一问题的核心技术思路是实现“分级存储”(Memory Tiering),将高频访问的“热数据”驻留在...