GKE で単一の GPU を持つモデルを提供する

Autopilot Standard

このチュートリアルでは、Google Kubernetes Engine（GKE）モードで GPU を使用して単純なモデルを提供する方法について説明します。このチュートリアルでは、単一の L4 Tensor Core GPU を使用する GKE クラスタを作成し、オンライン推論を行うように GKE インフラストラクチャを準備します。このチュートリアルでは、オンラインサービングでよく使用される次の 2 つのフレームワークを使用します。

目標

このチュートリアルは、GKE クラスタで事前トレーニング済みの ML モデルをホストするインフラストラクチャエンジニア、MLOps エンジニア、DevOps エンジニア、クラスタ管理者を対象としています。

このチュートリアルでは、次の手順について説明します。

GKE Autopilot クラスタまたは GKE Standard クラスタを作成する。
事前トレーニング済みモデルが存在する Cloud Storage バケットを構成する。
選択したオンライン推論フレームワークをデプロイする。
デプロイされたサービスにテストリクエストを送信する。

費用

このチュートリアルでは、課金対象である次の Google Cloud コンポーネントを使用します。

GKE
Cloud Storage
L4 GPU アクセラレータ
下り（外向き）トラフィック

料金計算ツールを使うと、予想使用量に基づいて費用の見積もりを出すことができます。

このチュートリアルを終了した後、作成したリソースを削除すると、それ以上の請求は発生しません。詳細については、クリーンアップをご覧ください。

始める前に

プロジェクトを設定する

Sign in to your Google Cloud account. If you're new to Google Cloud, create an account to evaluate how our products perform in real-world scenarios. New customers also get $300 in free credits to run, test, and deploy workloads.

Google Cloud Console のプロジェクトセレクタのページで、[プロジェクトを作成] をクリックして新しい Google Cloud プロジェクトの作成を開始します。

プロジェクトセレクタに移動

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

GKE API を有効にします。

API を有効にする

Google Cloud Console のプロジェクトセレクタのページで、[プロジェクトを作成] をクリックして新しい Google Cloud プロジェクトの作成を開始します。

プロジェクトセレクタに移動

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

GKE API を有効にします。

API を有効にする

Google Cloud CLI のデフォルト値を設定する

Google Cloud コンソールで、Cloud Shell インスタンスを起動します。
Cloud Shell を開く

このサンプルアプリのソースコードをダウンロードします。

git clone https://github.com/GoogleCloudPlatform/kubernetes-engine-samples
cd kubernetes-engine-samples/ai-ml/gke-online-serving-single-gpu

デフォルトの環境変数を設定します。
```
gcloud config set project PROJECT_ID
gcloud config set compute/region COMPUTE_REGION
```
次の値を置き換えます。
- PROJECT_ID: Google Cloud のプロジェクト ID。
- COMPUTE_REGION: 使用するアクセラレータタイプをサポートする Compute Engine のリージョン（例: L4 GPU の場合は us-central1）。

Cloud Shell で、次の環境変数を作成します。

export PROJECT_ID=$(gcloud config get project)
export REGION=$(gcloud config get compute/region)
export K8S_SA_NAME=gpu-k8s-sa
export GSBUCKET=$PROJECT_ID-gke-bucket
export MODEL_NAME=mnist
export CLUSTER_NAME=online-serving-cluster

GKE クラスタを作成する

GKE Autopilot クラスタまたは GKE Standard クラスタの単一 GPU でモデルを提供できます。フルマネージドの Kubernetes エクスペリエンスを実現するには、Autopilot クラスタを使用することをおすすめします。GKE Autopilot では、リソースはモデルのリクエストに基づいて自動的にスケーリングされます。

ワークロードに最適な GKE の運用モードを選択するには、GKE の運用モードを選択するをご覧ください。

Autopilot

次のコマンドを実行して、GKE Autopilot クラスタを作成します。

  gcloud container clusters create-auto ${CLUSTER_NAME} \
      --region=${REGION} \
      --project=${PROJECT_ID} \
      --release-channel=rapid

GKE は、デプロイされたワークロードからのリクエストに応じた CPU ノードと GPU ノードを持つ Autopilot クラスタを作成します。

Standard

次のコマンドを実行して、GKE Standard クラスタを作成します。

  gcloud container clusters create ${CLUSTER_NAME} \
    --project=${PROJECT_ID}  \
    --region=${REGION}  \
    --workload-pool=${PROJECT_ID}.svc.id.goog \
    --addons GcsFuseCsiDriver \
    --release-channel=rapid \
    --num-nodes=1

クラスタの作成には数分かかることもあります。

次のコマンドを実行してノードプールを作成します。

  gcloud container node-pools create gpupool \
    --accelerator type=nvidia-l4,count=1,gpu-driver-version=latest \
    --project=${PROJECT_ID} \
    --location=${REGION} \
    --node-locations=${REGION}-a \
    --cluster=${CLUSTER_NAME} \
    --machine-type=g2-standard-8 \
    --num-nodes=1

GKE は、ノードごとに 1 つの L4 GPU を含む単一のノードプールを作成します。

Cloud Storage バケットを作成する

提供する事前トレーニング済みモデルを保存する Cloud Storage バケットを作成します。

Cloud Shell で次のコマンドを実行します。

gcloud storage buckets create gs://$GSBUCKET

GKE 用 Workload Identity 連携を使用してバケットにアクセスするようにクラスタを構成する

クラスタから Cloud Storage バケットにアクセスできるようにする手順は次のとおりです。

Google Cloud サービスアカウントを作成する。
クラスタに Kubernetes ServiceAccount を作成する。
Kubernetes ServiceAccount を Google Cloud サービスアカウントにバインドする。

Google Cloud サービスアカウントを作成する

Google Cloud コンソールで、[サービスアカウントの作成] ページに移動します。

[サービスアカウントの作成] に移動
[サービスアカウント ID] フィールドに「gke-ai-sa」と入力します。
[作成して続行] をクリックします。
[ロール] リストで、[Cloud Storage] > [Storage Insights コレクタサービス] ロールを選択します。
[ 別のロールを追加] をクリックします。
[ロールを選択] リストで、[Cloud Storage] > [Storage Object Admin] ロールを選択します。
[続行] をクリックして [完了] をクリックします。

クラスタに Kubernetes ServiceAccount を作成する

Cloud Shell で、次の操作を行います。

Kubernetes Namespace を作成します。

kubectl create namespace gke-ai-namespace

Namespace に Kubernetes ServiceAccount を作成します。

kubectl create serviceaccount gpu-k8s-sa --namespace=gke-ai-namespace

Kubernetes ServiceAccount を Google Cloud サービスアカウントにバインドする

Cloud Shell で、次のコマンドを実行します。

Google Cloud サービスアカウントに IAM バインディングを追加します。

gcloud iam service-accounts add-iam-policy-binding gke-ai-sa@PROJECT_ID.iam.gserviceaccount.com \
    --role roles/iam.workloadIdentityUser \
    --member "serviceAccount:PROJECT_ID.svc.id.goog[gke-ai-namespace/gpu-k8s-sa]"

--member フラグは、Google Cloud の Kubernetes ServiceAccount の完全な ID を指定します。

Kubernetes ServiceAccount にアノテーションを付けます。

kubectl annotate serviceaccount gpu-k8s-sa \
    --namespace gke-ai-namespace \
    iam.gke.io/gcp-service-account=gke-ai-sa@PROJECT_ID.iam.gserviceaccount.com

オンライン推論サーバーをデプロイする

各オンライン推論フレームワークは、特定の形式の事前トレーニング済み ML モデルを探すように構成されています。次のセクションでは、使用するフレームワークに応じて推論サーバーをデプロイする方法について説明します。

Triton

Cloud Shell で、事前トレーニング済み ML モデルを Cloud Storage バケットにコピーします。
```
gsutil cp -r src/triton-model-repository gs://$GSBUCKET
```

Kubernetes にフレームワークをデプロイします。

envsubst < src/gke-config/deployment-triton.yaml | kubectl --namespace=gke-ai-namespace apply -f -

フレームワークがデプロイされたことを確認します。

kubectl get deployments --namespace=gke-ai-namespace

フレームワークの準備が整うと、出力は次のようになります。

NAME                 READY   UP-TO-DATE   AVAILABLE   AGE
triton-deployment    1/1     1            1           5m29s

Deployment にアクセスする Service をデプロイします。

kubectl apply --namespace=gke-ai-namespace -f src/gke-config/service-triton.yaml

外部 IP が割り当てられていることを確認します。

kubectl get services --namespace=gke-ai-namespace

出力は次のようになります。

NAME            TYPE           CLUSTER-IP       EXTERNAL-IP     PORT(S)                                        AGE
kubernetes      ClusterIP      34.118.224.1     <none>          443/TCP                                        60m
triton-server   LoadBalancer   34.118.227.176   35.239.54.228   8000:30866/TCP,8001:31035/TCP,8002:30516/TCP   5m14s

EXTERNAL-IP 列にある triton-server の IP アドレスをメモします。

Service と Deployment が正常に機能していることを確認します。
```
curl -v EXTERNAL_IP:8000/v2/health/ready
```
出力は次のようになります。
```
...
< HTTP/1.1 200 OK
< Content-Length: 0
< Content-Type: text/plain
...
```

TF Serving

Cloud Shell で、事前トレーニング済み ML モデルを Cloud Storage バケットにコピーします。
```
gsutil cp -r src/tfserve-model-repository gs://$GSBUCKET
```

Kubernetes にフレームワークをデプロイします。

envsubst < src/gke-config/deployment-tfserve.yaml | kubectl --namespace=gke-ai-namespace apply -f -

フレームワークがデプロイされたことを確認します。

kubectl get deployments --namespace=gke-ai-namespace

フレームワークの準備が整うと、出力は次のようになります。

NAME                 READY   UP-TO-DATE   AVAILABLE   AGE
tfserve-deployment   1/1     1            1           5m29s

Deployment にアクセスする Service をデプロイします。

kubectl apply --namespace=gke-ai-namespace -f src/gke-config/service-tfserve.yaml

外部 IP が割り当てられていることを確認します。

kubectl get services --namespace=gke-ai-namespace

出力は次のようになります。

NAME            TYPE           CLUSTER-IP       EXTERNAL-IP     PORT(S)                                        AGE
kubernetes      ClusterIP      34.118.224.1     <none>          443/TCP                                        60m
tfserve-server  LoadBalancer   34.118.227.176   35.239.54.228   8500:30003/TCP,8000:32194/TCP                  5m14s

EXTERNAL-IP 列にある tfserve-server の IP アドレスをメモします。

Service と Deployment が正常に機能していることを確認します。

curl -v EXTERNAL_IP:8000/v1/models/mnist

EXTERNAL_IP は、外部 IP アドレスに置き換えます。

出力は次のようになります。

...
< HTTP/1.1 200 OK
< Content-Type: application/json
< Date: Thu, 12 Oct 2023 19:01:19 GMT
< Content-Length: 154
<
{
  "model_version_status": [
        {
        "version": "1",
        "state": "AVAILABLE",
        "status": {
          "error_code": "OK",
          "error_message": ""
        }
      }
    ]
}

モデルを提供する

Triton

Cloud Shell で Python 仮想環境を作成します。

python -m venv ./mnist_client
source ./mnist_client/bin/activate

必要な Python パッケージをインストールします。
```
pip install -r src/client/triton-requirements.txt
```
画像を読み込んで Triton Inference Server をテストします。
```
cd src/client
python triton_mnist_client.py -i EXTERNAL_IP -m mnist -p ./images/TEST_IMAGE.png
```
次のように置き換えます。
- EXTERNAL_IP: 外部 IP アドレス。
- TEST_IMAGE: テストする画像のファイル名。src/client/images に保存されている画像を使用できます。
使用する画像に応じて、出力は次のようになります。
```
Calling Triton HTTP Service      ->      Prediction result: 7
```

TF Serving

Cloud Shell で Python 仮想環境を作成します。

python -m venv ./mnist_client
source ./mnist_client/bin/activate

必要な Python パッケージをインストールします。
```
pip install -r src/client/tfserve-requirements.txt
```

いくつかの画像を使用して TensorFlow Serving をテストします。

cd src/client
python tfserve_mnist_client.py -i EXTERNAL_IP -m mnist -p ./images/TEST_IMAGE.png

次のように置き換えます。

EXTERNAL_IP: 外部 IP アドレス。
TEST_IMAGE: 0～9 の値です。src/client/images に保存されている画像を使用できます。

使用する画像に応じて、次のような出力が表示されます。

  Calling TensorFlow Serve HTTP Service    ->      Prediction result: 5

クリーンアップ

このガイドで作成したリソースについて Google Cloud アカウントに課金されないようにするには、次のいずれかを行います。

GKE クラスタを保持する場合: クラスタにある Kubernetes リソースと Google Cloud リソースを削除する
Google Cloud プロジェクトを保持する場合: GKE クラスタと Google Cloud リソースを削除する
プロジェクトを削除する

クラスタにある Kubernetes リソースと Google Cloud リソースを削除する

Kubernetes Namespace とデプロイしたワークロードを削除します。

Triton

kubectl -n gke-ai-namespace delete -f src/gke-config/service-triton.yaml
kubectl -n gke-ai-namespace delete -f src/gke-config/deployment-triton.yaml
kubectl delete namespace gke-ai-namespace

TF Serving

kubectl -n gke-ai-namespace delete -f src/gke-config/service-tfserve.yaml
kubectl -n gke-ai-namespace delete -f src/gke-config/deployment-tfserve.yaml
kubectl delete namespace gke-ai-namespace

Cloud Storage バケットを削除します。
1. [バケット] ページに移動します。
  
  [バケット] に移動
2. PROJECT_ID-gke-bucket のチェックボックスをオンにします。
3. [削除] をクリックします。
4. 削除を確定するには、「DELETE」と入力して [削除] をクリックします。
Google Cloud サービスアカウントを削除する
1. [サービスアカウント] ページに移動します。
  
  [サービスアカウント] に移動
2. プロジェクトを選択します。
3. gke-gpu-sa@PROJECT_ID.iam.gserviceaccount.com のチェックボックスをオンにします。
4. [削除] をクリックします。
5. 削除を確定するには、[削除] をクリックします。

GKE クラスタと Google Cloud リソースを削除する

GKE クラスタを削除します。
1. [クラスタ] ページに移動します。
  
  [クラスタ] に移動
2. online-serving-cluster のチェックボックスをオンにします。
3. [削除] をクリックします。
4. 削除を確定するには、「online-serving-cluster」と入力して [削除] をクリックします。
Cloud Storage バケットを削除します。
1. [バケット] ページに移動します。
  
  [バケット] に移動
2. PROJECT_ID-gke-bucket のチェックボックスをオンにします。
3. [削除] をクリックします。
4. 削除を確定するには、「DELETE」と入力して [削除] をクリックします。
Google Cloud サービスアカウントを削除する
1. [サービスアカウント] ページに移動します。
  
  [サービスアカウント] に移動
2. プロジェクトを選択します。
3. gke-gpu-sa@PROJECT_ID.iam.gserviceaccount.com のチェックボックスをオンにします。
4. [削除] をクリックします。
5. 削除を確定するには、[削除] をクリックします。

プロジェクトを削除する

注意: プロジェクトを削除すると、次のような影響があります。

プロジェクト内のすべてのものが削除されます。このドキュメントのタスクで既存のプロジェクトを使用した場合、それを削除すると、そのプロジェクトで行った他の作業もすべて削除されます。
カスタムプロジェクト ID が失われます。このプロジェクトを作成したときに、将来使用するカスタムプロジェクト ID を作成した可能性があります。そのプロジェクト ID を使用した URL（たとえば、appspot.com）を保持するには、プロジェクト全体ではなくプロジェクト内の選択したリソースだけを削除します。

複数のアーキテクチャ、チュートリアル、クイックスタートを実施する予定がある場合は、プロジェクトを再利用すると、プロジェクトの割り当て上限を超えないようにすることができます。

In the Google Cloud console, go to the Manage resources page.
Go to Manage resources
In the project list, select the project that you want to delete, and then click Delete.
In the dialog, type the project ID, and then click Shut down to delete the project.

GKE で単一の GPU を持つモデルを提供する

目標

費用

始める前に

プロジェクトを設定する

Google Cloud CLI のデフォルト値を設定する

GKE クラスタを作成する

Autopilot

Standard

Cloud Storage バケットを作成する

GKE 用 Workload Identity 連携を使用してバケットにアクセスするようにクラスタを構成する

Google Cloud サービス アカウントを作成する

クラスタに Kubernetes ServiceAccount を作成する

Kubernetes ServiceAccount を Google Cloud サービス アカウントにバインドする

オンライン推論サーバーをデプロイする

Triton

TF Serving

モデルを提供する

Triton

TF Serving

クリーンアップ

クラスタにある Kubernetes リソースと Google Cloud リソースを削除する

Triton

TF Serving

GKE クラスタと Google Cloud リソースを削除する

プロジェクトを削除する

次のステップ

Google Cloud サービスアカウントを作成する

Kubernetes ServiceAccount を Google Cloud サービスアカウントにバインドする