PyTorch 1.9发布!移动端疯狂更新,网友:我的最爱( 二 )


PyTorch 1.9发布!移动端疯狂更新,网友:我的最爱
本文插图
分布式训练
TorchElastic是PyTorch的一个核心功能 , 它能够让用户在抢占式实例上运行分布式训练 。
PyTorch 1.9发布!移动端疯狂更新,网友:我的最爱
本文插图
△TorchElastic运行原理
新版本中 , 添加了一个基于c10d::Store的“独立”集合点 , 能够在本地支持弹性、容错分布式训练 。
此外 , 现在还可在RPC中支持CUDA , 并支持对分布式训练进行分析等 。
PyTorch Profiler
PyTorch Profiler是用来对PyTorch模型性能分析的一个工具 。
通过可视化页面来帮助我们分析具体运行情况 。
PyTorch 1.9发布!移动端疯狂更新,网友:我的最爱
本文插图
在1.9版本中 , 可以在Windows和Mac中支持新的torch.profiler API了 。
新的API支持现有分析器功能 , 可以与CUPTI库集成(仅限Linux) , 追踪设备上的CUDA内核 , 并为长期运行项目提供支持 , 例如:
PyTorch 1.9发布!移动端疯狂更新,网友:我的最爱
本文插图
【PyTorch 1.9发布!移动端疯狂更新,网友:我的最爱】