Imagens do SO da TPU

Ao criar uma VM de TPU usando o Compute Engine, você pode usar uma imagem de sistema operacional (SO) pública gerenciada pelo Google ou uma imagem personalizada. As imagens gerenciadas pelo Google são otimizadas para a performance da TPU e incluem os drivers e agentes necessários. Por padrão, todos os Google Cloud projetos têm acesso a essas imagens públicas do SO.

Imagens do SO gerenciadas pelo Google

O Google oferece imagens do SO baseadas no Ubuntu que são pré-configuradas com o ambiente de execução, os drivers e os agentes da TPU necessários para que as cargas de trabalho funcionem.

A tabela a seguir lista as imagens do SO gerenciadas pelo Google para cada versão de TPU compatível:

Versão da TPU Versão do SO Projeto de imagem Família de imagens
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

Imagens do SO personalizadas

Com o Compute Engine, é possível usar uma imagem do SO personalizada para VMs de TPU. As imagens personalizadas são úteis para:

  • Pré-carregar softwares e bibliotecas personalizados para reduzir o tempo de inicialização da VM.
  • Usar uma distribuição ou versão do SO que não esteja incluída na lista gerenciada pelo Google.
  • Aplicar modificações personalizadas do kernel ou configurações de segurança.

Modelo de responsabilidade compartilhada

O uso de uma imagem do SO personalizada segue um modelo de responsabilidade compartilhada entre você e o Google, conforme mostrado na tabela a seguir:

Recurso Responsabilidade do Google Responsabilidade do usuário
Agentes e drivers de TPU Publicar modificações e liberar agentes Instalar e configurar agentes e drivers
Teste de compatibilidade N/A Realizar testes de compatibilidade e performance
Manutenção do SO N/A Gerenciar atualizações e licenciamento do SO

Para mais informações sobre como criar e usar sua própria imagem do Ubuntu, consulte Usar uma imagem do SO personalizada.

A seguir