Mantenha tudo organizado com as coleções Salve e categorize o conteúdo com base nas suas preferências.

Perguntas frequentes: Cloud TPU

Neste documento, você verá uma lista de perguntas frequentes sobre as Cloud TPUs. Ela é dividida em seções:

  1. Perguntas frequentes sobre o framework independente: perguntas sobre o uso de Cloud TPUs, independentemente do framework de ML que você esteja usando.
  2. Perguntas frequentes sobre o JAX: perguntas sobre o uso de Cloud TPUs com o JAX.
  3. Perguntas frequentes sobre o PyTorch: perguntas sobre como usar Cloud TPUs com PyTorch.

Perguntas frequentes sobre a estrutura independente

Como posso verificar qual processo está usando a TPU em uma VM da Cloud TPU?

Execute sudo lsof -w /dev/accel* na VM da Cloud TPU para imprimir o ID do processo e outras informações sobre o processo usando a TPU.

Como adiciono um volume de disco permanente a uma VM da Cloud TPU?

Para ver mais informações, consulte Adicionar um disco permanente a uma VM da TPU.

Quais opções de armazenamento são compatíveis/recomendados para treinamento com uma VM de TPU?

Para mais informações, consulte Opções de armazenamento do Cloud TPU.

Perguntas frequentes sobre o JAX

Como saber se a TPU está sendo usada pelo meu programa?

Há duas maneiras de verificar se o JAX está usando a TPU:

  1. Use a função jax.devices(). Exemplo:

    assert jax.devices()[0].platform == 'tpu'
    
  2. Criar um perfil do programa e verificar se o perfil contém operações de TPU. Para saber mais, consulte Como criar o perfil de programas JAX.

Para mais informações, consulte Perguntas frequentes do JAX

Perguntas frequentes sobre PyTorch

Como saber se a TPU está sendo usada pelo meu programa?

É possível executar os seguintes comandos Python:

>>> import torch_xla.core.xla_model as xm
>>> xm.get_xla_supported_devices(devkind="TPU")

Verifique se você consegue ver algum dispositivo TPU.