繼去年年底 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 並獲取更多詳細信息。