开发者社区 问答 正文

在数据并行的场景下,ZeRO技术是如何帮助减少单卡显存占用的?

在数据并行的场景下,ZeRO技术是如何帮助减少单卡显存占用的?image.png

展开
收起
萝卜丝丸子 2024-08-05 21:46:56 96 分享 版权
1 条回答
写回答
取消 提交回答
  • 在数据并行的场景下,ZeRO技术通过将optimizer state和gradient分片存储在不同的卡上,从而减少了单卡的persistent memory占用,有效解决了模型变大时容易超出单卡显存限制的问题。

    2024-08-05 22:08:25
    赞同 7 展开评论
问答地址: