PyTorch 2.0正式版发布！一行代码提速2倍，100%向后兼容(1/7)

NVIDIA A100 GPU eager mode torch.compile 针对不同模型的提速表示据PyTorch基金会称，新编译器在运用Float32精度方式时运转速度进步了21%，在运用自动混合精度（AMP）方式时运转速度进步了51%。在这163个模型中，torch.compile能够在93%模型上正常运转。「在PyTorch 2.x的道路图中，我们希望在性能和可扩展性方面让编译方式越走越远。有一些工作还没有开端。有些工作由于带宽不够而措施 ... [查看原文]

标签： 积家

全部图片

查看原图