Usage
Description
Wrap docker run and inject W&B environment variables automatically.
Pass all arguments through to docker run while injecting:
-
WANDB_API_KEY: Inject the current API key if logged in so the
container can authenticate with W&B.
-
WANDB_DOCKER: Inject the resolved image ID if the image can be detected
from the arguments so W&B can track which Docker image produced the run.
Add --runtime nvidia automatically if nvidia-docker is detected
on the host and --runtime is not already set.
For example, to run the Python script train.py inside the “my-image” container:
Arguments
Last modified on September 30, 2026
