Miután a PyTorch Foundation tavaly év vége óta beszélt a PyTorch 2.0-ról, a mai napon hivatalosan is kiadják a PyTorch 2.0-t. A PyTorch 2.0 jelentős optimalizálással „feltölti” a jobb teljesítményt a CPU és a GPU üzemmódban egyaránt.
A PyTorch 2.0 fő API-ként a torch/compile-t használja, a TorchInductor NVIDIA és AMD GPU-kkal mostantól az OpenAI Tritonra támaszkodik a mély tanulási fordítóhoz, a Metal Performance Shaders háttérrendszer pedig GPU-gyorsított PyTorch-ot biztosít macOS-en, gyorsabb következtetési teljesítményt AWS Graviton CPU-k és számos más új prototípus-szolgáltatás és technológia. A gyorsabb CPU-működés érdekében számos „kritikus” optimalizálás szükséges a GNN-következtetéshez és a betanításhoz, valamint a gyorsabb CPU-teljesítmény az Intel oneDNN Graph használatával.
Letöltések és további részletek a PyTorch 2.0-ról a PyTorch.org webhelyen.
