TFX中的GPU和分布式培训

Question

TFX使用Apache.Beam运行ML Pipeline。

我想将TFX Pipelines部署在某些机器上具有GPU设备的Flink群集上，怎么做：

1）确保Trainer组件在具有GPU设备的机器上运行？

2）部署分布式Trainer组件？

谢谢。