PyTorch
Article mis en ligne le 9 juillet 2020

La bibliothèque PyTorch est accessible au travers du module python/3.6.8

- Environnement Python + CUDA et environnement virtuel pytorch

- Lancement de PyTorch avec Python

Vérifier que son installation est fonctionnelle

Vous pouvez utiliser le script ci-joint pur vérifier que pytorch fonctionne correctement sur les GPUs :

  • Copiez le script dans un répertoire sur Olympe
  • Suivez les instructions en début de script (en Anglais)