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.
