Apache Beam で PyTorch を動かしてみた¶

最近の Apache Beam のアップデートで推論のランタイムがサポートされた。TensorFlow や PyTorch、TensorRT、sk-learn などが動く。Google Cloud のドキュメントを参考に Apache Beam で PyTorch を動かしてみた。Vertex AI Workbench の環境で動かし、公式のドキュメントをなぞるだけで最後まで動いた。Dataflow 上で動かすとこまでできなかったので時間がある時に動かしてみたい。