Changement de version du logiciel sur votre Cloud TPU

Aperçu

Pour utiliser Cloud TPU, vous devez sélectionner une version logicielle correspondant au logiciel exécuté sur votre VM locale. Cette version logicielle peut désormais être activée sur un Cloud TPU en cours d'exécution, sans supprimer ni recréer le TPU. Cela permet également de configurer Cloud TPU avec des versions nocturnes spécifiques des frameworks logiciels. Il est toujours recommandé de sélectionner une version compatible de ces frameworks.

Utilisation

La méthode recommandée pour changer de version consiste à utiliser la bibliothèque Python cloud-tpu-client.

Exemple d'utilisation de TensorFlow.

import tensorflow as tf
from cloud_tpu_client import Client

c = Client()
c.configure_tpu_version(tf.__version__, restart_type='ifNeeded')

Cela configurera le Cloud TPU pour qu'il corresponde à la version exacte de TensorFlow exécutée sur votre VM locale, y compris les versions officielles et les veilleuses datées.

La bibliothèque communique directement avec Cloud TPU. Ce code doit donc être exécuté sur une VM du même réseau. Nous vous recommandons de l'exécuter dans le code pour le reste de votre modèle.

Options logicielles supplémentaires

Tensorflow inclut une chaîne tf.__version__ qui constitue le moyen le plus simple de configurer la bonne version. Les autres options logicielles sont les suivantes:

  • PyTorch - pytorch-0.1, pytorch-0.5-dev20200320, pytorch-nightly
  • Jax - tpu_driver, tpu_driver0.1-dev20200320, tpu_driver_nightly

Par exemple, pour configurer un TPU à exécuter avec la dernière version nocturne de PyTorch.

from cloud_tpu_client import Client
c = Client()
c.configure_tpu_version('pytorch-nightly', restart_type='ifNeeded')