继去年年底 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 的更多详细信息。