首先是官网链接 https://pytorch.org/get-started/pytorch-2.0/
然后是对应的中文翻译: https://zhuanlan.zhihu.com/p/588854132
代码和上手的中文版本 https://zhuanlan.zhihu.com/p/631616829
在今天的 PyTorch 2022 开发者大会上,PyTorch 团队发布了一个新特性torch.compile
,这个新特性将 PyTorch 的性能推向了新高度,并开始将 PyTorch 的部分实现从 C++ 中迁移到 Python 中。他们相信这是 PyTorch 一个实质性的新方向–因此称之为 PyTorch 2.0。torch.compile
是一个完全附加的(和可选的)功能,因此 PyTorch 2.0 100% 向后兼容, 因此基于 PyTorch 1.x 开发的项目可以不用做任何修改就能迁移到 PyTorch2.0 。
一行提速 30%
只需使用torch.compile
对模型进行编译,就可以享受 PyTorch 2.0 给模型速度带来的极致提升
发表回复