Sortie de PyTorch 1.8

 (pytorch.org)
3 points par tkwlsrl 2021-03-14 | 2 commentaires | Partager sur WhatsApp

Le 4 mars 2021, PyTorch 1.8 a été publié.

  • Ajout de nouvelles API et mise à jour des fonctionnalités d’apprentissage distribué

  • Prise en charge de ROCm pour les GPU AMD (bêta)

  • Des mises à jour sont également proposées pour TorchVision, TorchText et TorchAudio

  • Tutoriels PyTorch Mobile fournis (exécution de PyTorch sur Android / iOS)

À lire aussi

2 commentaires

 
tkwlsrl 2021-03-14

Je lis toujours vos actualités avec intérêt. J’en laisse une moi aussi.

À titre d’information, PyTorch 1.8 a ajouté une fonctionnalité de memory fraction qui permet de partager la mémoire GPU.

https://pytorch.org/docs/stable/cuda.html

torch.cuda.set_per_process_memory_fraction(fraction, device=None)
 
xguru 2021-03-14

Merci~ ! Je ne suis pas très calé en machine learning haha

C’est sympa de voir ça résumé comme ça. N’hésitez pas à en refaire de temps en temps à l’avenir ! ^^