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 proyectos Google Cloud tienen acceso a estas imágenes de SO públicas.

Imágenes de SO administradas por Google

Google proporciona imágenes del 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 del SO administradas por Google para cada versión de TPU compatible:

Versión de TPU Versión del SO Versión de kernel Proyecto de imagen Familia de imágenes
TPU7x (Ironwood) Ubuntu 24.04 LTS 6.17 ubuntu-os-accelerator-images ubuntu-accel-2404-amd64-tpu-tpu7x
v6e (Trillium), v5p Ubuntu 22.04 LTS 6.8 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:

  • Precarga de software y bibliotecas personalizados para reducir el tiempo de inicio de la VM
  • Usar una distribución o versión del SO que no se incluye 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 Publica modificaciones y lanza agentes Instala y configura agentes y controladores
Pruebas de compatibilidad N/A Realiza pruebas de compatibilidad y rendimiento
Mantenimiento del SO N/A Administra 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?