Berikutan Yayasan PyTorch membincangkan PyTorch 2.0 sejak akhir tahun lepas, hari ini menandakan keluaran PyTorch 2.0 secara rasmi dihantar. PyTorch 2.0 mempunyai pengoptimuman yang ketara untuk”mengecas”nya dengan prestasi yang lebih baik untuk kedua-dua mod operasi CPU dan GPU.
PyTorch 2.0 menggunakan torch/compile sebagai API utamanya, TorchInductor dengan NVIDIA dan GPU AMD kini bergantung pada OpenAI Triton untuk pengkompil pembelajaran mendalam, bahagian belakang Metal Performance Shaders menyediakan PyTorch dipercepatkan GPU pada macOS, prestasi inferens yang lebih pantas pada CPU AWS Graviton, dan pelbagai ciri dan teknologi prototaip baharu yang lain. Untuk operasi CPU yang lebih pantas juga terdapat beberapa pengoptimuman”kritikal”kepada inferens dan latihan GNN serta prestasi CPU yang lebih pantas dengan menggunakan Intel oneDNN Graph.
Muat turun dan butiran lanjut tentang PyTorch 2.0 melalui PyTorch.org.