Expérimental : Horovod
Article mis en ligne le 20 novembre 2020
dernière modification le 10 décembre 2020
Horovod est un framework permettant d’entraîner des modèles d’intelligence artificielle en utilisant plusieurs GPUs.
Version installée
Le module intègre python 3.6, tensorflow 2.3.0 et horovod :
Version d’Horovod | 0.20.3 |
---|---|
Version de python | 3.6.12 |
Version de tensorflow | 2.3.0 |
Si vous souhaitez utiliser Horovod avec d’autres applications d’intelligence artificeille, contactez-nous !
Utilisation et exemples
Pour utiliser horovod :
Des exemples de scripts python sont disponibles dans le répertoire $HOROVOD_EXAMPLES
Le script suivant permet de faire tourner l’exemple tensorflow2_keras_synthetic_benchmark.py
Pendant l’exécution de l’exemple, n’oubliez pas de vérifier que les quatre GPUs sont effectivement utilisés avec la commande :