Pytorch分布式训练

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


文章目录


1、并行训练

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

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

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

相关推荐
YisquareTech2 分钟前
如何实现智能补货?EDI与ERP集成打造零售库存的“自动闭环”
大数据·人工智能·零售·伊士格科技·erp集成
观远数据6 分钟前
数据驱动零售新生态:观远BI打造终端经营“透视镜”
大数据·人工智能·信息可视化·数据分析·零售
思通数科人工智能大模型7 分钟前
零售场景下的数智店商:解决盗损问题,化解隐性成本痛点
人工智能·目标检测·计算机视觉·数据挖掘·知识图谱·零售
世优科技虚拟人17 分钟前
数字刘邦“亮相”汉文化论坛:AI数字人推动文化传承与科技融合
人工智能·科技·数字人·文化创新·文化传承
今天吃饺子22 分钟前
如何用MATLAB调用python实现深度学习?
开发语言·人工智能·python·深度学习·matlab
飞哥数智坊1 小时前
Anthropic 出手,大幅降低 MCP 的 Token 消耗
人工智能·claude·mcp
贝多财经1 小时前
千里科技报考港股上市:高度依赖吉利,AI智驾转型收入仍为零
大数据·人工智能·科技
嵌入式-老费1 小时前
自己动手写深度学习框架(pytorch入门)
人工智能·pytorch·深度学习
irisMoon061 小时前
yolov5单目测距+速度测量+目标跟踪
人工智能·yolo·目标跟踪
Linux猿1 小时前
365科技简报 2025年11月13日 星期四
人工智能·科技简报