pyTorch2.0学习上手合集

首先是官网链接 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.0torch.compile 是一个完全附加的(和可选的)功能,因此 PyTorch 2.0 100% 向后兼容, 因此基于 PyTorch 1.x 开发的项目可以不用做任何修改就能迁移到 PyTorch2.0 

一行提速 30%

只需使用torch.compile对模型进行编译,就可以享受 PyTorch 2.0 给模型速度带来的极致提升


已发布

分类

来自

标签:

评论

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注