PyTorch 2.10 on välja antud

PyTorch — populaarne masinõppe raamistik — on uuendatud versioonile 2.10.

Peamiste uuenduste seas on:

• AMD ROCm toetamine:

  • Sisse viidud rühmitatud GEMM toetus.
  • ROCm-i tugi Windowsile on paranenud.
  • Uued GPU mudelid GFX1150/GFX1151 (AI 300 seeria) on lisatud hipblaslt toetuse nimekirja.
  • Funktsionaalsused on laienenud, sealhulgas scaled_mm v2 ja AOTriton scaled_dot_product_attention tugi.

• Intel GPU tugi:

  • Uued Torch XPU API-d on sisse viidud Inteli jaoks.
  • ATen'i täiendavate operaatorite tugi.
  • Intel GPU jaoks on optimeeritud jõudlus.

• NVIDIA CUDA tugi:

  • Laienenud võimalused malli tuumade kirjutamiseks.
  • CUDA 13 toetus on paranenud.

• Python 3.14 tugi torch.compile() jaoks ning katsetus Python 3.14 ehitamiseks ilma tõlkija globaalsete lukustusteta (free-threaded).

• Tuumade käitamise ülekande kulude vähenemine tänu Torch Inductor'i horisontaalsetele kombinatsioonide liitmisele.

Muudatuste täielik nimekiri on saadaval github.

Valmis variandid, mis on kompileeritud erinevatele GPU-dele, operatsioonisüsteemidele ja keeledele (Python/C++/Java), on loetletud Pytorch veebisaidil

Allikas: linux.org.ru