Utilisation
Description
Encapsule docker run et injecte automatiquement les variables d’environnement W&B.
Transmet tous les arguments à docker run en injectant :
-
WANDB_API_KEY : injecte la clé API actuelle si vous êtes connecté, afin que le
conteneur puisse s’authentifier auprès de W&B.
- WANDB_DOCKER : injecte l’ID d’image résolu si l’image peut être déduite
des arguments, afin que W&B puisse identifier l’image Docker ayant produit le run.
Ajoute automatiquement --runtime nvidia si nvidia-docker est détecté
sur l’hôte et que --runtime n’est pas déjà défini.
Par exemple, pour exécuter le script Python train.py dans le conteneur « my-image » :
Arguments
Dernière modification le 30 septembre 2026
