Cloud TPU 上のソフトウェア バージョンの切り替え

概要

Cloud TPU を使用するには、ローカル VM で実行されているソフトウェアと一致するソフトウェア バージョンを選択する必要があります。このソフトウェア バージョンは、TPU を削除して再作成せずに、実行中の Cloud TPU で切り替えられるようになりました。そのため、特定の夜間バージョンのソフトウェア フレームワークで Cloud TPU を構成することもできます。これらのフレームワークのサポートされているバージョンを選択することをおすすめします。

用途

バージョンを切り替えるには、cloud-tpu-client Python ライブラリを使用することをおすすめします。

TensorFlow の使用例

import tensorflow as tf
from cloud_tpu_client import Client

c = Client()
c.configure_tpu_version(tf.__version__, restart_type='ifNeeded')

これにより、ローカル VM で実行されている TensorFlow の正確なバージョンと一致するように Cloud TPU が構成されます。これには、公式リリースと日付の付いた夜間バージョンが含まれます。

このライブラリは Cloud TPU と直接通信します。そのため、このコードは同じネットワーク内の VM で実行する必要があります。残りのモデルのコード内で実行することをおすすめします。

その他のソフトウェア オプション

Tensorflow には、正しいバージョンを構成する最も簡単な方法である tf.__version__ 文字列が含まれています。その他のソフトウェア オプションは次のとおりです。

  • PyTorch - pytorch-0.1pytorch-0.5-dev20200320pytorch-nightly
  • Jax - tpu_drivertpu_driver0.1-dev20200320tpu_driver_nightly

たとえば、PyTorch の最新の夜間ビルドで実行するように TPU を構成します。

from cloud_tpu_client import Client
c = Client()
c.configure_tpu_version('pytorch-nightly', restart_type='ifNeeded')