PyTorch 2.0 现已出货,具有更好的 CPU 和 GPU 性能

173次阅读
没有评论

继去年年底 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 2.0 现已出货,具有更好的 CPU 和 GPU 性能

通过PyTorch.org下载 PyTorch 2.0 的更多详细信息。

正文完
 
Windows12系统管理员
版权声明:本站原创文章,由 Windows12系统管理员 2023-03-16发表,共计365字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。
评论(没有评论)