1.2 为什么选择PyTorch 2.0

在PyTorch Conference 2022上,PyTorch官方正式发布了PyTorch 2.0,整场活动含“compiler”率极高,跟先前的1.x版本相比,2.0版本有了颠覆式的变化。

PyTorch 2.0中发布了大量足以改变PyTorch使用方式的新功能,它提供了相同的Eager模式和用户体验,同时通过torch.compile增加了一个编译模式,在训练和推理过程中可以对模型进行加速,从而提供更佳的性能以及对Dynamic Shapes和Distributed的支持。