Vue d'ensemble
PyTorch 2.4, publié le 25 juillet 2024, ajoute le support officiel de Python 3.12 et poursuit les améliorations de torch.compile.
Fonctionnalités principales
Python 3.12
PyTorch supporte officiellement Python 3.12, avec des roues précompilées disponibles sur PyPI pour toutes les plateformes.
python
import torch
import sys
print(f'Python {sys.version}')
print(f'PyTorch {torch.__version__}')
print(f'CUDA disponible : {torch.cuda.is_available()}')
# torch.compile fonctionne avec Python 3.12
@torch.compile
def calcul(x):
return torch.nn.functional.gelu(x) * 2
print(calcul(torch.randn(4)))