> ## Documentation Index
> Fetch the complete documentation index at: https://wb-21fd5541-run-filter-ui-updates.mintlify.site/llms.txt
> Use this file to discover all available pages before exploring further.

# How to make W&B Launch work with TensorFlow on GPU

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:

```json theme={null}
{
    "gpus": "all",
    "builder": {
        "accelerator": {
            "base_image": "tensorflow/tensorflow:latest-gpu"
        }
    }
}
```

<Note>
  In versions before W\&B 0.15.6, use `cuda` instead of `accelerator` as the parent key for `base_image`.
</Note>