Usage
Description
Wrapdocker 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.
--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
| Name | Default | Type |
|---|---|---|
docker_run_args | STR | False |