继去年年底 PyTorch 基金会大谈 PyTorch 2.0 之后,今天标志着 PyTorch 2.0 正式发布。PyTorch 2.0 进行了重大优化,可以为 CPU 和 GPU 操作模式“增压”提供更好的性能。
PyTorch 2.0 使用 torch/compile 作为其主要 API,带有 NVIDIA 和 AMD GPU 的 TorchInductor 现在依赖 OpenAI Triton 作为深度学习编译器,Metal Performance Shaders 后端在 macOS 上提供 GPU 加速的 PyTorch,在 AWS Graviton CPU 上更快的推理性能,以及各种其他新的原型功能和技术。为了更快的 CPU 操作,还对 GNN 推理和训练进行了多项“关键”优化,并通过使用英特尔 oneDNN Graph 提高了 CPU 性能。
通过PyTorch.org下载 PyTorch 2.0 的更多详细信息。
原创文章,作者:校长,如若转载,请注明出处:https://www.yundongfang.com/Yun220168.html