Can I specify a Dockerfile and let W&B build a Docker image for me?

  less than a minute  

This feature suits projects with stable requirements but frequently changing codebases.

After configuring the Dockerfile, specify it in one of three ways to W&B:

  • Use Dockerfile.wandb
  • Use W&B CLI
  • Use W&B App

Include a Dockerfile.wandb file in the same directory as the W&B run’s entrypoint. W&B utilizes this file instead of the built-in Dockerfile.

Use the --dockerfile flag with the wandb launch command to queue a job:

wandb launch --dockerfile path/to/Dockerfile
Bash

When adding a job to a queue in the W&B App, provide the Dockerfile path in the Overrides section. Enter it as a key-value pair with "dockerfile" as the key and the path to the Dockerfile as the value.

The following JSON demonstrates how to include a Dockerfile in a local directory:

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

Last modified February 6, 2025: 8b20a01