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)