Passer au contenu principal

Utilisation

docker [OPTIONS] [DOCKER_RUN_ARGS]... [DOCKER_IMAGE]

Description

Exécutez du code dans un conteneur Docker avec W&B configuré. Démarrez un conteneur Docker, injectez la variable d’environnement WANDB_DOCKERs et montez le répertoire de travail actuel dans /app par défaut. Injectez WANDB_API_KEY si vous êtes connecté. Redéfinissez le point d’entrée du conteneur pour garantir que wandb est installé. Passez des arguments supplémentaires pour les ajouter à docker run avant le nom de l’image. Utilisez une image par défaut si aucune n’est spécifiée. Utilisez --jupyter pour installer et démarrer JupyterLab sur le port 8888. Activez automatiquement le runtime NVIDIA si NVIDIA Docker est disponible sur l’hôte. Docker doit être installé et en cours d’exécution sur la machine hôte. Pour injecter des variables d’environnement W&B dans une commande docker run existante sans modifier le point d’entrée, utilisez wandb docker-run. Par exemple, pour exécuter l’image par défaut et monter un jeu de données dans le conteneur :
wandb docker -v /mnt/dataset:/app/data
Pour exécuter l’image par défaut et démarrer JupyterLab :
wandb docker -v /mnt/dataset:/app/data --jupyter
Pour exécuter une image avec prise en charge du GPU à l’aide d’une commande d’entraînement :
wandb docker wandb/deepo:keras-gpu --no-tty --cmd "python train.py"

Arguments

NomPar défautType
docker_run_argsSTRFalse
docker_imageSTRFalse

Options

OptionTypeDescription
--nvidiaIndicateur booléenUtilise le runtime nvidia ; la valeur par défaut est nvidia si nvidia-docker est présent. Par défaut: False
--digestIndicateur booléenAffiche le digest de l’image, puis quitte. Par défaut: False
--jupyterIndicateur booléenExécute JupyterLab dans le conteneur. Par défaut: False
--dirSTRRépertoire dans lequel monter le code dans le conteneur. Par défaut: /app
--no-dirIndicateur booléenNe monte pas le répertoire actuel. Par défaut: False
--shellSTRShell avec lequel démarrer le conteneur. Par défaut: /bin/bash
--portSTRPort hôte auquel lier Jupyter. Par défaut: 8888
--cmdSTRCommande à exécuter dans le conteneur. Par défaut: None
--no-ttyIndicateur booléenExécute la commande sans tty. Par défaut: False