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 :