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
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"
}
Feedback
Was this page helpful?
Glad to hear it! Please tell us how we can improve.
Sorry to hear that. Please tell us how we can improve.