Passer au contenu principal

Documentation Index

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

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

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 : 
{
    "gpus": "all",
    "builder": {
        "accelerator": {
            "base_image": "tensorflow/tensorflow:latest-gpu"
        }
    }
}
Dans les versions antérieures à W&B 0.15.6, utilisez cuda au lieu de accelerator comme clé parente de base_image.