プライベートモデルハブを作成する

プライベートハブを作成する

1. SageMaker Python SDK をインストールし、必要な Python パッケージをインポートします。

   # Install the SageMaker Python SDK
   !pip3 install sagemaker --force-reinstall --quiet
   
   # Import the necessary Python packages
   import boto3
   from sagemaker import Session
   from sagemaker.jumpstart.hub.hub import Hub

2. SageMaker AI セッションを初期化します。

   sm_client = boto3.client('sagemaker')
   session = Session(sagemaker_client=sm_client)
   session.get_caller_identity_arn()

3. 内部ハブ名、UI 表示名、UI ハブの説明など、プライベートハブの詳細情報を設定します。

   注記

   ハブの作成時に Amazon S3 バケット名を指定しない場合、SageMaker AI ハブサービスはユーザーに代わって新しいバケットを作成します。新しいバケットの名前の構造は sagemaker-hubs-REGION-ACCOUNT_ID です。

   HUB_NAME="Example-Hub"
   HUB_DISPLAY_NAME="Example Hub UI Name"
   HUB_DESCRIPTION="A description of the example private curated hub."
   REGION="us-west-2"

4. 管理者 IAM ロールに、プライベートハブを作成するために必要な Amazon S3 アクセス許可があることを確認します。ロールに必要なアクセス許可がない場合は、IAM コンソールの [ロール] ページに移動します。管理者ロールを選択し、[アクセス許可ポリシー] ペインで [許可を追加] を選択して、JSON エディタを使用して次のアクセス許可を持つインラインポリシーを作成します。

   {
       "Version": "2012-10-17",
       "Statement": [
           {
               "Action": [
                   "s3:ListBucket",
                   "s3:GetObject",
                   "s3:GetObjectTagging"
               ],
               "Resource": [
                   "arn:aws:s3:::jumpstart-cache-prod-REGION",
                   "arn:aws:s3:::jumpstart-cache-prod-REGION/*"
               ],
               "Effect": "Allow"
           }
       ]
   }

5. hub.create() を使用して、手順 3 の設定情報でプライベートモデルハブを作成します。

   hub = Hub(hub_name=HUB_NAME, sagemaker_session=session)
   
   try:
   # Create the private hub
     hub.create(
         description=HUB_DESCRIPTION,
         display_name=HUB_DISPLAY_NAME
     )
     print(f"Successfully created Hub with name {HUB_NAME} in {REGION}")
   # Check that no other hubs with this internal name exist
   except Exception as e:
     if "ResourceInUse" in str(e):
       print(f"A hub with the name {HUB_NAME} already exists in your account.")
     else:
       raise e

6. 次の describe コマンドを使用して、新しいプライベートハブの設定を確認します。

   hub.describe()