wandb dockeris a command that starts a docker container, passes in wandb environment variables, mounts your code, and ensures wandb is installed. By default the command uses a docker image with TensorFlow, PyTorch, Keras, and Jupyter installed. You can use the same command to start your own docker image:
wandb docker my/image:latest. The command mounts the current directory into the "/app" directory of the container, you can change this with the "--dir" flag.
wandb docker-runcommand is provided for production workloads. It's meant to be a dropin replacement for
nvidia-docker. It's a simple wrapper to the
docker runcommand that adds your credentials and the WANDB_DOCKER environment variable to the call. If you do not pass the "--runtime" flag and
nvidia-dockeris available on the machine, this also ensures the runtime is set to nvidia.
wandb restore username/project:run_idwill checkout a new branch restoring your code then launch the exact docker image used for training pre-populated with the original command.