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

Google Cloud のPyTorch 推論のノートブックを Vertex AI Workbench 上で実行し、Apache Beam で PyTorch の推論を最後まで動かせた。Dataflow 上での実行は確認できていないので、時間があるときに試したい。

Apache Beam のアップデートで推論のランタイムがサポートされ、PyTorch のほか、TensorFlow、TensorRT、scikit-learn なども動くようになった。