多卡训练神经网络训练集 batch_size=x 内存分配问题
·
如果一张卡batch_size=1可以跑
那样才有机会去使用多卡训练
不然batch_size=1都不能训练成功,那么再怎么分配都不行
————————————————————————————————————————
如果batch_size=8分配到8张卡去跑,那么8张卡每张占1个batch_size=1去跑,
如果batch_size=8分配到7张卡去跑,主卡会占2个batch_size=1去跑,那么剩下7张卡每张占1个batch_size=1去跑,
然后在主卡中汇总数据,所以说:
1、多卡训练对于数据的分配不是智能的,是主卡分最多,其他卡占平均。
2、主卡中负担是最大的,各种数据和框架都要在这里面。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)