GPU и Распределить Обучение в TFX

Question

TFX использует Apache.Beam для запуска ML Pipeline.

Я хочу развернуть мои конвейеры TFX в кластере Flink с устройствами GPU на некоторых машинах, как я могу:

1) Убедитесь, что компонент Trainer работает на машинах с устройствами с графическим процессором?

2) развернуть распределенный компонент Trainer?

Благодарю.