Skip to main content

Documentation Index

Fetch the complete documentation index at: https://wb-21fd5541-docs-2632.mintlify.app/llms.txt

Use this file to discover all available pages before exploring further.

To run TensorFlow jobs on GPUs with W&B Launch, specify a custom base image for the container build. This lets the resulting container access the GPU. Add an image tag under the builder.accelerator.base_image key in the resource configuration. For example: 
{
    "gpus": "all",
    "builder": {
        "accelerator": {
            "base_image": "tensorflow/tensorflow:latest-gpu"
        }
    }
}
In versions before W&B 0.15.6, use cuda instead of accelerator as the parent key for base_image.