Oossa

PyTorch 2.15 rende predefinito il nuovo backend NCCL

La release introduce strumenti per diagnosticare i blocchi nell’addestramento distribuito e aggiornamenti per hardware Apple, AMD, Intel e NVIDIA.

BreveDi Pubblicato da Oossa: 1 min di lettura

PyTorch ha rilasciato la versione 2.15, impostando come predefinito il backend NCCL riscritto per i processi che usano NCCL, il sistema di comunicazione che coordina le GPU NVIDIA. Se un gruppo di GPU si blocca durante l’addestramento, un nuovo flight recorder può registrare le operazioni in corso su ciascun rank, aiutando i team a indagare sul problema.

La release introduce anche trasferimenti di dati unidirezionali tramite un backend NIXL e offre agli sviluppatori un maggiore controllo sulla memoria e sulle comunicazioni nell’addestramento distribuito. Tra le altre novità ci sono kernel di inferenza più veloci per Apple Silicon, CUDA 13.2 come versione predefinita dei pacchetti wheel e il supporto a Python 3.15 in torch.compile.

Perché conta

I team che addestrano modelli su più GPU NVIDIA possono contare su registrazioni diagnostiche in caso di blocchi nelle comunicazioni, anche se le note di rilascio non quantificano quanto possano ridurre i tempi di ripristino.

Questo articolo ti è stato utile?
Condividi

Leggi anche

Oossa · Newsletter

La settimana dell'IA, spiegata

Ogni lunedì: le notizie che contano, in parole semplici. Gratis, niente spam.