Monitorizar con la CLI tpu-info
La CLI tpu-info es una herramienta para detectar dispositivos TPU de Cloud y leer métricas del tiempo de ejecución de la biblioteca libtpu, como el uso de memoria y el ciclo de trabajo. Admite capturas estáticas y únicas, así como la transmisión en directo para monitorizar las métricas de forma continua.
Instalación
Instala la última versión con pip:
pip install tpu-info
También puedes instalar tpu-info desde la fuente:
pip install git+https://github.com/google/cloud-accelerator-diagnostics/#subdirectory=tpu_info
Si ya has instalado una versión de tpu-info, asegúrate de que sea compatible con tu entorno y de que no falten métricas ni funciones.
Para obtener más información, consulta Faltan funciones o métricas.
Acceder a métricas estándar de LibTPU mediante la CLI
Usa el siguiente comando para ver las métricas tpu-info predeterminadas con la CLI:
tpu-info
El resultado debería ser similar al siguiente:
TPU Chips
┏━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━┳━━━━━━━━━┳━━━━━━━━┓
┃ Chip ┃ Type ┃ Devices ┃ PID ┃
┡━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━╕━━━━━━━━━╕━━━━━━━━┩
│ /dev/vfio/0 │ TPU v6e chip │ 1 │ 1052 │
│ /dev/vfio/1 │ TPU v6e chip │ 1 │ 1052 │
│ /dev/vfio/2 │ TPU v6e