Python: Google Colaboratory で Cloud TPU を TensorFlow から試してみる - CUBE SUGAR CONTAINER

Google Colaboratory では、ランタイムのタイプを変更することで Cloud TPU (Tensor Processing Unit) を利用できる。 Cloud TPU は、Google が開発しているハードウェアアクセラレータの一種。 利用することで、行列計算のパフォーマンス向上が期待できる。 ただ、Cloud TPU は CPU や GPU に比べると扱う上でのクセがそれなりにつよい。 今回は、そんな Cloud TPU を使ってみることにする。 使った環境は次のとおり。 # pip list | grep "^tensorflow " tensorflow 2.1.0 もくじ もくじ 下準備 TPU に接続する 単一のデバイスで演算する 複数のデバイスで演算する 単一のデバイスで勾配降下法を試す 複数のデバイスで CNN を tf.keras で学習する 参考 下準