Passer au contenu principal

Documentation Index

Fetch the complete documentation index at: https://wb-21fd5541-john-wbdocs-2044-rename-serverless-products.mintlify.app/llms.txt

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

Pour les jobs TensorFlow utilisant des GPU, spécifiez une image de base personnalisée pour la construction du conteneur. Cela garantit une bonne utilisation des GPU pendant les runs. 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.