GPU con Dataflow

Las GPU de Dataflow proporcionan los beneficios acelerados directamente a su canalización de tratamiento de datos por lotes o de transmisión. Usa Dataflow para simplificar el proceso de obtención de datos a la GPU y el aprovechamiento de la localidad de datos. Al mismo tiempo, obtén todos los beneficios del sistema completamente administrado de Dataflow: el aprovisionamiento de host, el ajuste de escala automático, la tolerancia a errores y mucho más.
Sigue este instructivo para ver cómo compilar una imagen de contenedor personalizada y ejecutar una canalización de Dataflow con GPU.
Sigue este instructivo para compilar una canalización de PyTorch y ejecutarla en Dataflow con GPU.
El tipo de GPU L4 es útil para ejecutar canalizaciones de inferencia de aprendizaje automático.
Sigue este instructivo para compilar una canalización de TensorFlow y ejecutarla en Dataflow con GPU.

Recursos

Observa un ejemplo de un flujo de trabajo de desarrollador para compilar canalizaciones con GPU.
Usa el ajuste adecuado en tus trabajos por lotes para personalizar los recursos de los trabajadores y reducir los costos.
Si tienes problemas para ejecutar tu trabajo de Dataflow con GPU, sigue estos pasos de solución de problemas.