Pytorch分布式训练

现在深度学习模型占用显存大,数据量也大,单张显卡上训练已经满足不了要求了,只有多GPU并行训练才能加快训练速度;并行训练又分为模型并行和数据并行两种。模型并行比较少用到,这里主要介绍数据并行,pytorch中数据并行有两种DataParallel和DistributedDataParallel,前者是pytorch训练早期采用的,由于其单线程和显存利用率低等缺点,现在大多使用后者。


文章目录


1、并行训练

并行训练分两种,模型并行和数据并行。

1)模型并行。模型并行通常是指要训练的模型非常大,大到一块卡根本放不下,因而需要把模型进行拆分放到不同的卡上。例如早期的AlexNet就是拆分模型利用两块GPU训练的。

2)数据并行。数据并行通常用于训练数据非常庞大的时候,比如有几百万张图像用于训练模型。此时,如果只用一张卡来进行训练,那么训练时间就会非常的长。或者模型比较大,由于单卡显存的限制,训练时的batch size不能设置过大。这时就需要多个GPU训练来提升batchsize大小。

相关推荐
阿正的梦工坊4 小时前
WebArena:一个真实的网页环境,用于构建更强大的自主智能体
人工智能·深度学习·机器学习·大模型·llm
大模型实验室Lab4AI4 小时前
小模型驱动大智能:NVIDIA 新研究实现 2.5 倍效率提升,成本直降 70%
人工智能
Hcoco_me4 小时前
Word2Vec 核心知识点速记版
人工智能·自然语言处理·word2vec
qijiabao41134 小时前
深度学习|可变形卷积DCNv3编译安装
人工智能·python·深度学习·机器学习·cuda
风途知识百科4 小时前
小型车载自动气象站
人工智能·数码相机
TonyLee0174 小时前
卷积操作记录(pytorch)
人工智能·pytorch·深度学习
小鸡吃米…4 小时前
机器学习所需技能
人工智能·机器学习
小马过河R4 小时前
ReAct和Function Calling之间的纠葛与恩恩怨怨
人工智能·语言模型·agent·智能体
HySpark4 小时前
基于声纹预处理技术的智能语音识别实现
人工智能·语音识别
l1m0_4 小时前
UI设计规范工程化,AI生成Ant Design设计稿流程拆解
人工智能·ui·产品经理·设计·arco design·设计规范