家からでもTPUを使いたい - I'm chizuchizu

家からでもTPUを使いたい GCPでTPUをJupyter Lab上で動かしたい人向けの記事です。作業 TPUインスタンスを作る export PROJECT_ID=hogehoge export TPU_NAME=hogehoge TPU_NAMEという環境変数が無いとTPUを動かすことができないので最低でもこれだけは定義しておかないといけません。 gcloudを使ってローカルから作る。TF-versionを選択しなかったら最新版の2.4が選択されるが、GCPが対応してるのは2.3.1までらしく、エラーが出てしまうので明示的に 2.3.1としておきましょう。（この仕様 is 何）また、プリエンプティブルインスタンスにしたいときはTPUとマシンそれぞれ引数が必要とのこと。 ctpu up --project=$PROJECT_ID \ --zone=us-central1-b \ --tf-version=2.3.1 \ --name=TPU_NAME --preemptible \ --preemptible-vm これで、インスタンスが立ち上がります。

1 mentions: