微比恩 > 信息聚合 > PyTorch 2.0 现已发布:编译器性能大幅提升,100% 向后兼容

PyTorch 2.0 现已发布:编译器性能大幅提升,100% 向后兼容

2023-03-19 15:33:28来源: IT之家

IT之家 3 月 19 日消息,PyTorch 2.0 稳定版现已发布。跟先前 1.0 版本相比,2.0 有了颠覆式的变化。在 PyTorch 2.0 中,最大的改进主要是 API 的 torch.compile,新编译器比先前「eager mode」所提供的即时生成代码的速度快得多,性能得以进一步提升。IT之家附官网地址:https://pytorch.org/GitHub 地址:https://github.com/pytorch/pytorch/releases新版本更新之处包括稳定版的 Accelerated Transformers(以前称为 Better Transformers);Beta 版包括作为 PyTorch 2.0 主要 API 的 torch.compile、作为 torch.nn.functional 一部分的 scaled_dot_product_attention函数、MPS 后端、to

关注公众号