Usa una imagen de SO personalizada

Puedes usar una imagen de SO personalizada para tus VMs de TPU y precargar software, usar una distribución de SO específica o aplicar modificaciones personalizadas del kernel. Crear una imagen personalizada implica realizar modificaciones específicas del sistema durante el proceso de creación de la imagen y configurar la imagen para que controle las tareas de tiempo de arranque necesarias para la funcionalidad de la TPU.

Ten en cuenta las siguientes renuncias de responsabilidad si usas una imagen de SO personalizada con TPU:

  • Google proporciona imágenes predeterminadas de Ubuntu con asistencia a largo plazo (LTS) optimizadas para TPU. Los cambios en el SO que se indican en esta página solo se validan para las imágenes de Ubuntu LTS optimizadas para TPU y compatibles con Google.
  • Eres responsable de extrapolar los cambios necesarios del SO para cualquier otra distribución del SO o imágenes personalizadas. Google no garantiza que las modificaciones para Ubuntu que se indican en esta página funcionen con otras distribuciones del SO o con otra imagen de Ubuntu con un kernel personalizado.
  • Google no compila ni proporciona pruebas para ninguna imagen de SO que no sean las imágenes predeterminadas de Ubuntu LTS optimizadas para TPU. Debes compilar y probar tu imagen de SO personalizada.

Para obtener más información sobre las imágenes de Ubuntu LTS optimizadas para TPU predeterminadas, consulta Imágenes del SO de TPU.

Requisitos previos

Tu imagen base debe tener instalados los siguientes componentes:

  • Python 3
  • gcloud CLI

Realiza modificaciones durante la creación de imágenes

Aplica las siguientes modificaciones mientras compilas tu imagen personalizada de Ubuntu.

Vincula dispositivos de TPU a VFIO

Para permitir que el SO invitado acceda al hardware de la TPU, debes vincular los dispositivos de TPU al controlador vfio-pci.

  1. Crea un archivo de reglas udev llamado 99-tpu-vfiopci.rules en /etc/udev/rules.d/:

    # Rules for binding vfio-enabled TPU devices to vfio-pci.

# v5p
SUBSYSTEM=="pci", ACTION==