【AI编译器之PyTorch】系列01篇:PyTorch 2.0新特性开撸!图模式打开了潘多拉! 它来了它来了!PyTorch 2.0 中发布了大量足以改变 PyTorch 使用方式的新功能,提供相同 eager mode 和用户体验,同时通过 torch.compile 增加了一个编译模式,在训练和推理过程中可以对模型进行加速,从而提供更佳的性能和对 Dynamic Shapes 及分布式运行的支持!让我们一起撸一下PyTorch提供的新特性,了解下PyTorch 2.0 的图模式吧!
www.bilibili.com