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)