更快、更 Pythonic 的 PyTorch 2.0 | 非常值得期待

CSDN 叶庭云：https://yetingyun.blog.csdn.net/

---

一、前言

PyTorch 官方：原本是 1.14 来着，但我们这次的新特性太好用了，所以就直接叫 2.0 了。现在，只需添加一行代码，PyTorch 2.0 就能让你在训练 Transformer 模型时，实现 1.5-2 倍的速度提升！

先让 ChatGPT 简单介绍一下 PyTorch 2.0：

2022 年 12 月 2 日，PyTorch 2.0 正式发布！这次的更新不仅将 PyTorch 的性能推到了一个新的高度，同时也加入了对 dynamic shapes 和