PyTorch
Article mis en ligne le 9 juillet 2020
dernière modification le 24 septembre 2020

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

- Environnement Python + CUDA et environnement virtuel pytorch

    • pytorch-1.4
    • pytorch-1.6

- 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)