Compatibilidade do Dataflow com GPUs

Nesta página, você verá informações básicas sobre como as GPUs funcionam com o Dataflow, incluindo informações sobre pré-requisitos e tipos de GPU compatíveis.

O uso de GPUs em jobs do Dataflow permite acelerar algumas tarefas de processamento de dados. As GPUs podem executar determinados cálculos mais rapidamente do que as CPUs. Esses cálculos geralmente são álgebra numérica ou linear, geralmente usados em casos de uso de processamento de imagem e machine learning. A extensão da melhoria de desempenho varia de acordo com o caso de uso, o tipo de computação e a quantidade de dados processados.

Pré-requisitos para usar GPUs no Dataflow

Preços

Jobs que usam GPUs geram cobranças conforme especificado na página de preços do Dataflow.

Disponibilidade

Os seguintes tipos de GPU são compatíveis com o Dataflow:

Tipo de GPU worker_accelerator string
NVIDIA® L4 nvidia-l4
NVIDIA® A100 40 GB nvidia-tesla-a100
NVIDIA® A100 80 GB nvidia-a100-80gb
NVIDIA® Tesla® T4 nvidia-tesla-t4
NVIDIA® Tesla® P4 nvidia-tesla-p4
NVIDIA® Tesla® V100 nvidia-tesla-v100
NVIDIA® Tesla® P100 nvidia-tesla-p100

Para mais informações sobre cada tipo de GPU, incluindo dados de desempenho, consulte Plataformas de GPU do Compute Engine.

Para informações sobre regiões e zonas disponíveis para GPUs, consulte Disponibilidade de regiões e zonas de GPU na documentação do Compute Engine.

Cargas de trabalho recomendadas

A tabela a seguir fornece recomendações de qual tipo de GPU usar em diferentes cargas de trabalho. Os exemplos na tabela são apenas sugestões. Você precisa testar no seu ambiente para determinar o tipo de GPU apropriado para a carga de trabalho.

Para informações mais detalhadas sobre o tamanho da memória da GPU, a disponibilidade de recursos e os tipos de carga de trabalho ideais para diferentes modelos de GPU, consulte a tabela geral de comparação na página de plataformas de GPU.

Carga de trabalho A100 L4 T4
Ajuste de modelos Recomendações
Inferência de modelos grandes Recomendações Recomendações
Inferência de modelo médio Recomendações Recomendações
Inferência de modelo pequeno Recomendações Recomendações

A seguir