开发者社区> 问答> 正文

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

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

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

    2024-08-05 22:08:25
    赞同 7 展开评论 打赏
问答地址:
问答排行榜
最热
最新

相关电子书

更多
低代码开发师(初级)实战教程 立即下载
冬季实战营第三期:MySQL数据库进阶实战 立即下载
阿里巴巴DevOps 最佳实践手册 立即下载