图解分布式训练三——nn.parallel.DistributedDataParallel详解.pdf

图解分布式训练三——nn.parallel.DistributedDataParallel详解.pdf

图解分布式训练(三)——nn.parallel.DistributedDataParallel

来:AiGC面试宝典

为什么需要nn.parallel.DistributedDataParallel

多GPU并行训练的原理就是将模型参数和数据分布到多个GPU上,同时利用多个GPU计算加速训练过程。具体实

现需要考虑以下三个问题:

1.**数据如何划分?**因为模型需要处理的数据通常很大,将所有数据放入单个GPU内存中可能会导致内存不

足,因此我们需要将数据划分到多个GPU上。一般有两种划分方式:

•数据并行:将数据分割成多个小批次,每

文档评论(0)

1亿VIP精品文档

相关文档