Plusieurs versions de a bibliothèque pytorch sont installées

• 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
• pytorch-1.7 (installé par conda/4.9.2) module purge module load conda/4.9.2 conda activate pytorch-1.7

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)