TPU インスタンスを作成する

このドキュメントでは、TPU v6e（Trillium）仮想マシン（VM）インスタンスを作成し、接続して、計算を実行する方法について説明します。このクイックスタートで作成する TPU インスタンスは、デフォルトの消費オプションであるオンデマンドを使用します。

始める前に

  1. アカウントにログインします。 Google Cloud を初めて使用する場合は、 アカウントを作成して、 実際のシナリオでプロダクトがどのように機能するかを評価してください。 Google Cloud新規のお客様には、ワークロードの実行、テスト、デプロイができる無料クレジット $300 分を差し上げます。

  2. Google Cloud CLI をインストールします。

  3. 外部 ID プロバイダ（IdP）を使用している場合は、まず連携 ID を使用して gcloud CLI にログインする必要があります。

  4. gcloud CLI を初期化するには、次のコマンドを実行します: 

    gcloud init

  5. プロジェクトを作成または選択します Google Cloud 。

    プロジェクトを選択または作成するために必要なロール

    • プロジェクトを選択する: プロジェクトの選択には特定の IAM ロールは必要ありません。ロールが付与されているプロジェクトを選択できます。
    • プロジェクトを作成する: プロジェクトを作成するには、プロジェクト作成者ロール （roles/resourcemanager.projectCreator）が必要です。これには resourcemanager.projects.create 権限が含まれています。ロールを付与する方法を確認する。

    • プロジェクトを作成する: Google Cloud 

      gcloud projects create PROJECT_ID

      PROJECT_ID は、作成する Google Cloud プロジェクトの名前に置き換えます。

    • 作成した Google Cloud プロジェクトを選択します。

      gcloud config set project PROJECT_ID

      PROJECT_ID は、 Google Cloud プロジェクトの名前に置き換えます。

  6. このガイドで既存のプロジェクトを使用する場合は、 このガイドを完了するために必要な権限があることを 確認します。新しいプロジェクトを作成した場合は、 必要な権限がすでに付与されています。

  7. プロジェクトで課金が有効になっていることを確認します Google Cloud 。

  8. Compute Engine API を有効にします。

    API を有効にするために必要なロール

    API を有効にするには、 権限を含む Service Usage 管理者 IAM ロール（roles/serviceusage.serviceUsageAdmin）が必要です。serviceusage.services.enableロールを付与する方法を確認する。

    gcloud services enable compute.googleapis.com

  9. Google Cloud CLI をインストールします。

  10. 外部 ID プロバイダ（IdP）を使用している場合は、まず連携 ID を使用して gcloud CLI にログインする必要があります。

  11. gcloud CLI を初期化するには、次のコマンドを実行します: 

    gcloud init

  12. プロジェクトを作成または選択します Google Cloud 。

    プロジェクトを選択または作成するために必要なロール

    • プロジェクトを選択する: プロジェクトの選択には特定の IAM ロールは必要ありません。ロールが付与されているプロジェクトを選択できます。
    • プロジェクトを作成する: プロジェクトを作成するには、プロジェクト作成者ロール （roles/resourcemanager.projectCreator）が必要です。これには resourcemanager.projects.create 権限が含まれています。ロールを付与する方法を確認する。

    • プロジェクトを作成する: Google Cloud 

      gcloud projects create PROJECT_ID

      PROJECT_ID は、作成する Google Cloud プロジェクトの名前に置き換えます。

    • 作成した Google Cloud プロジェクトを選択します。

      gcloud config set project PROJECT_ID

      PROJECT_ID は、 Google Cloud プロジェクトの名前に置き換えます。

  13. このガイドで既存のプロジェクトを使用する場合は、 このガイドを完了するために必要な権限があることを 確認します。新しいプロジェクトを作成した場合は、 必要な権限がすでに付与されています。

  14. プロジェクトで課金が有効になっていることを確認します Google Cloud 。

  15. Compute Engine API を有効にします。

    API を有効にするために必要なロール

    API を有効にするには、 権限を含む Service Usage 管理者 IAM ロール（roles/serviceusage.serviceUsageAdmin）が必要です。serviceusage.services.enableロールを付与する方法を確認する。

    gcloud services enable compute.googleapis.com
    16.

必要なロール

TPU インスタンスを作成し、SSH を使用して接続してコマンドを実行するために必要な権限を取得するには、プロジェクトに対する次の IAM ロールを付与するよう管理者に依頼してください。

ロールの付与については、プロジェクト、フォルダ、組織へのアクセス権の管理をご覧ください。

必要な権限は、カスタム ロールや他の事前定義 ロールから取得することもできます。

TPU v6e（Trillium）インスタンスを作成する

ローカル ターミナルまたは Cloud Shellで次のコマンドを実行します。

  1. TPU v6e インスタンスを作成するには、次の gcloud compute instances create コマンドを実行します。

    gcloud compute instances create quickstart-tpu \
    --machine-type=ct6e-standard-4t \
    --zone=europe-west4-a \
    --image-project=ubuntu-os-accelerator-images \
    --image-family=ubuntu-accel-2204-amd64-tpu-v5e-v5p-v6e \
    --maintenance-policy=TERMINATE

  2. 必要に応じて、gcloud compute instances describe コマンドを使用して、インスタンスが実行されていることを確認します。

    gcloud compute instances describe quickstart-tpu --zone=europe-west4-a

    出力で、status フィールドを確認して、インスタンスが実行中か保留中かを確認します。

TPU インスタンスに接続する

TPU インスタンスには、 Google Cloud コンソールまたは gcloud CLI を使用して接続できます。

SSH を使用して VM に接続するには、次の gcloud compute ssh コマンドを実行します。

gcloud compute ssh quickstart-tpu --zone=europe-west4-a

JAX をインストールする

TPU インスタンスで次のコマンドを実行して、JAX フレームワークをインストールします。

pip install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html

システム チェックを実行する

TPU インスタンスで次のコマンドを実行して、JAX が TPU にアクセスできること、基本オペレーションを実行できることを確認します。

  1. Python 3 インタプリタを起動する

    python3

  2. JAX モジュールをインポートする

    >>> import jax

  3. 使用可能な TPU コアの数を表示する

    >>> jax.device_count()

    出力には、TPU コアの数が表示されます。コアの数は、使用している TPU のバージョンによって異なります。詳細については、TPU のバージョンをご覧ください。

計算を実行する

  1. 次のコマンドを実行して、2 つの数値を加算します。

    >>> jax.numpy.add(1, 1)

    出力例を以下に示します。

    >>> Array(2, dtype=int32, weak_type=True)

  2. Python インタプリタを終了する

    >>> 
exit()

クリーンアップ

このページで使用したリソースについて、アカウントに課金されないようにするには、リソースを含むプロジェクトを削除します。 Google Cloud Google Cloud

このページで使用したリソースについて、 Google Cloud アカウントに課金されないようにするには、 このページで使用したリソースについて、 gcloud compute instances delete コマンドを使用して TPU インスタンスを削除します:

gcloud compute instances delete quickstart-tpu --zone=europe-west4-a

次のステップ