Puis-je spécifier un Dockerfile et laisser W&B construire une image Docker pour moi ?

Oui. Vous pouvez fournir votre propre Dockerfile et laisser W&B Launch construire l’image pour votre run. Cela vous permet de contrôler l’environnement de build tout en utilisant Launch pour placer des jobs en file d’attente et les exécuter. Cette approche fonctionne bien pour les projets dont les exigences sont stables, mais dont le code change.

Formatez votre Dockerfile de manière à utiliser des montages. Pour plus de détails, voir la documentation sur les montages Docker.

Après avoir configuré le Dockerfile, indiquez-le à W&B de l’une des manières suivantes :

• Utilisez Dockerfile.wandb
• Utilisez la CLI W&B
• Utilisez W&B App

Dockerfile.wandb

Incluez un fichier Dockerfile.wandb dans le même répertoire que le point d’entrée du run W&B. W&B utilise ce fichier à la place du Dockerfile intégré.

W&B CLI

Pour mettre un job en file d’attente, utilisez l’option --dockerfile avec la commande wandb launch :

wandb launch --dockerfile path/to/Dockerfile

W&B App

Lorsque vous ajoutez un job à une file d’attente dans W&B App, fournissez le chemin du Dockerfile dans la section Overrides. Saisissez-le sous forme de paire clé-valeur, avec "dockerfile" comme clé et le chemin comme valeur.

Le JSON suivant montre comment inclure un Dockerfile à partir d’un répertoire local :

job Launch W&B App

{
  "args": [],
  "run_config": {
    "lr": 0,
    "batch_size": 0,
    "epochs": 0
  },
  "entrypoint": [],
  "dockerfile": "./Dockerfile"
}