wandb docker-run

​

Utilisation

docker-run [OPTIONS] [DOCKER_RUN_ARGS]...

​

Description

Encapsule docker run et injecte automatiquement les variables d’environnement W&B.

Transmet tous les arguments à docker run tout 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étectée à partir des arguments, afin que W&B puisse suivre quelle image Docker a produit l’exécution.

Ajoute --runtime nvidia automatiquement 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” :

wandb docker-run my-image python train.py

​

Arguments

Nom	Par défaut	Type
docker_run_args	STR	False