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

 Environnement Python + CUDA et environnement virtuel pytorch

• • pytorch-1.4 module purge module load python/3.6.8 cuda/10.1.105 source activate pytorch-1.4
  • pytorch-1.6 module purge module load python/3.6.8 cuda/10.1.105 source activate 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)