Imágenes de SO de TPU

Cuando creas una VM de TPU con Compute Engine, puedes usar una imagen de sistema operativo (SO) pública administrada por Google o una imagen personalizada. Las imágenes administradas por Google están optimizadas para el rendimiento de la TPU y contienen los controladores y agentes necesarios. De forma predeterminada, todos los Google Cloud proyectos tienen acceso a estas imágenes públicas de SO.

Imágenes de SO administradas por Google

Google proporciona imágenes de SO basadas en Ubuntu que están preconfiguradas con el entorno de ejecución, los controladores y los agentes de TPU necesarios para que funcionen tus cargas de trabajo.

En la siguiente tabla, se enumeran las imágenes de SO administradas por Google para cada versión de TPU compatible:

Versión de TPU Versión del SO Proyecto de imagen Familia de imágenes
TPU7x (Ironwood) Ubuntu 24.04 LTS ubuntu-os-accelerator-images ubuntu-accel-2404-amd64-tpu-tpu7x
TPU v6e (Trillium) Ubuntu 22.04 LTS ubuntu-os-accelerator-images ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e
TPU v5p Ubuntu 22.04 LTS ubuntu-os-accelerator-images ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e

Imágenes personalizadas de SO

Con Compute Engine, puedes usar una imagen de SO personalizada para las VMs de TPU. Las imágenes personalizadas son útiles para lo siguiente:

  • Precargar software y bibliotecas personalizados para reducir el tiempo de inicio de la VM
  • Usar una distribución o versión del SO que no esté incluida en la lista administrada por Google
  • Aplicar modificaciones personalizadas del kernel o configuraciones de seguridad

Modelo de responsabilidad compartida

El uso de una imagen de SO personalizada sigue un modelo de responsabilidad compartida entre tú y Google, como se muestra en la siguiente tabla:

Función Responsabilidad de Google Responsabilidad del usuario
Agentes y controladores de TPU Publicar modificaciones y agentes de lanzamiento Instalar y configurar agentes y controladores
Pruebas de compatibilidad N/A Realizar pruebas de compatibilidad y rendimiento
Mantenimiento del SO N/A Administrar las actualizaciones y las licencias del SO

Para obtener más información sobre cómo compilar y usar tu propia imagen de Ubuntu, consulta Usa una imagen de SO personalizada.

¿Qué sigue?