Imagens do SO da TPU

Ao criar uma VM de TPU usando o Compute Engine, é possível 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 o desempenho da TPU e incluem os drivers e agentes necessários. Por padrão, todos os projetos do Google Cloud têm acesso a essas imagens públicas do SO.

Imagens do SO gerenciadas pelo Google

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

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

Versão da TPU Versão do SO Versão do kernel Projeto de imagem Família de imagens
TPU7x (Ironwood) Ubuntu 24.04 LTS 6.8 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

Imagens do SO personalizadas

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

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

As VMs TPU7x exigem uma versão mínima do kernel do Linux do SO convidado de 6.8.

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 lançar agentes Instalar e configurar agentes e drivers
Teste de compatibilidade N/A Realizar testes de compatibilidade e desempenho
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