PyTorch XLAでTPUを操作する - のんびりしているエンジニアの日記皆さんこんにちは お元気でしょうか。最近は宅配スーパーによりますます外出しなくなっています。 本日はTPUをPyTorchで使ってみます。 GPUと比較して、TPUは汎用性をなくした代わりによりDeepLearningに必要な演算を高速にできるようにしたものです。 Why TPU? TPUとは？ どの程度高速なのか どうやって使うのか PyTorch-XLA 準備 インストール 実装 インストール部 結果可視化 学習 最後に Why TPU? TPUとは？ TPUはニューラルネットワークの演算専用のアーキテクチャです。 一言で使うモチベーションをお伝えするのであれば「速いから」の一言です 他の用途には利用できない分、大規模な乗算と加算を高速に演算できます。 詳細は次のサイトを見ていただくのが良いと思います。 www.atmarkit.co.jp cloud.google.com どの程度高
