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

# Comment faire fonctionner W&B Launch avec TensorFlow sur GPU

Pour exécuter des jobs TensorFlow sur des GPU avec W\&B Launch, spécifiez une image de base personnalisée pour la construction du conteneur. Cela permet au conteneur résultant d’accéder au GPU. Ajoutez un tag d’image sous la clé `builder.accelerator.base_image` dans la configuration des ressources. Par exemple :

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

<Note>
  Dans les versions antérieures à W\&B 0.15.6, utilisez `cuda` au lieu de `accelerator` comme clé parente de `base_image`.
</Note>