Se usó la API de Cloud Translation para traducir esta página.
Switch to English

Cambia las versiones de software en tu Cloud TPU

Descripción general

Usar Cloud TPU requiere que se seleccione una versión de software para que coincida con el software que se ejecuta en tu VM local. Esta versión de software ahora se puede cambiar en una Cloud TPU en ejecución sin necesidad de borrar y volver a crear la TPU. Esto también permite configurar Cloud TPU con versiones nocturnas específicas de marcos de trabajo de software. Aún se recomienda seleccionar una versión compatible de estos marcos de trabajo.

Uso

La forma recomendada de cambiar de versión es usar la biblioteca de Python cloud-tpu-client de Python.

Ejemplo de uso para TensorFlow.

import tensorflow as tf
from cloud_tpu_client import Client

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

Esto configurará Cloud TPU para que coincida con la versión exacta de TensorFlow que se ejecuta en tu VM local, incluidos los lanzamientos oficiales y nocturnos.

La biblioteca se comunica directamente con Cloud TPU, por lo que este código debe ejecutarse en una VM en la misma red. Se recomienda ejecutarlo dentro del código para el resto de tu modelo.

Opciones adicionales de software

TensorFlow incluye una string de tf.__version__, que es la forma más sencilla de configurar la versión correcta. Otras opciones de software incluyen las siguientes:

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

Por ejemplo, para configurar una TPU para que se ejecute con la compilación nocturna más reciente de PyTorch.

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