Exemples de DTensor GCP

Ce projet contient des exemples d'utilisation de DTensor multi-client sur GCP avec un Un cluster de GPU ou de TPU.

Principaux

  1. Environnement gcloud sur la console locale :
gcloud auth login ...
gcloud config set project  ...
  1. Un bucket GCS que le compte de service GCE peut écrire. à démo checkpointing. définir le nom des chemins de préfix avec
export GCS_BUCKET=<bucket_name>

ou modifier bootstrap.sh.

Pour les conservateurs

Parce que cela nécessite tf-nightly, à jour périodique des exigences.txt dans Chaque catégorie à la version « bonne » connue pour ces exemples.