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

Para usar Cloud TPU, debes seleccionar una versión del software 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 borrar ni volver a crear la TPU. Esto también permite configurar Cloud TPU con versiones específicas de marcos de trabajo de software específicos. Todavía se recomienda seleccionar una versión compatible con estos marcos de trabajo.

Uso

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

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 de software adicionales

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

  • 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 a fin de que se ejecute con la última compilación nocturna de PyTorch.

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