カスタム SageMaker イメージを作成する

で Amazon SageMaker コンソールを開きますhttps://console.aws.amazon.com/sagemaker/。

左側のナビゲーションペインで、[管理設定]‭ を選択します。

[管理設定] で [イメージ] を選択します。

[カスタムイメージ] ページで [イメージの作成] を選択します。

イメージソース には、Amazon のコンテナイメージへのレジストリパスを入力しますECR。パスの形式は次のとおりです。

acct-id.dkr.ecr.region.amazonaws.com/repo-name[:tag] or [@digest]

[Next (次へ)] を選択します。

[イメージプロパティ] に次を入力します。

[送信] を選択します。

イメージと同じ行にある [バージョンを作成] を選択します。

イメージソース には、Amazon ECRコンテナイメージへのレジストリパスを入力します。コンテナイメージは、以前のバージョンのイメージと同じ SageMaker イメージであってはなりません。

SageMaker イメージを作成します。

aws sagemaker create-image \
    --image-name custom-image \
    --role-arn arn:aws:iam::<acct-id>:role/service-role/<execution-role>

レスポンスは次のようになります。

{
    "ImageArn": "arn:aws:sagemaker:us-east-2:acct-id:image/custom-image"
}

コンテナ SageMaker イメージからイメージバージョンを作成します。

aws sagemaker create-image-version \
    --image-name custom-image \
    --base-image <acct-id>.dkr.ecr.<region>.amazonaws.com/smstudio-custom:custom-image

レスポンスは次のようになります。

{
    "ImageVersionArn": "arn:aws:sagemaker:us-east-2:acct-id:image-version/custom-image/1"
}

イメージバージョンが正常に作成されたことを確認します。

aws sagemaker describe-image-version \
    --image-name custom-image \
    --version-number 1

レスポンスは次のようになります。

{
    "ImageVersionArn": "arn:aws:sagemaker:us-east-2:acct-id:image-version/custom-image/1",
    "ImageVersionStatus": "CREATED"
}

app-image-config-input.json という名前の設定ファイルを作成します。のName値は、この に関連付けられているイメージで kernelSpec 使用可能な の名前と一致するKernelSpecs必要がありますAppImageConfig。この値では、大文字と小文字が区別されます。コンテナ内のシェルjupyter-kernelspec listから を実行すると、イメージ kernelSpecs で使用可能な を見つけることができます。 MountPathは、Amazon Elastic File System (Amazon EFS) ホームディレクトリをマウントするためのイメージ内のパスです。Amazon EFSホームディレクトリがマウントされるとパスが上書きされるため、コンテナ内で使用するパスとは異なる必要があります。

{
    "AppImageConfigName": "custom-image-config",
    "KernelGatewayImageConfig": {
        "KernelSpecs": [
            {
                "Name": "python3",
                "DisplayName": "Python 3 (ipykernel)"
            }
        ],
        "FileSystemConfig": {
            "MountPath": "/home/sagemaker-user",
            "DefaultUid": 1000,
            "DefaultGid": 100
        }
    }
}

前のステップで作成した ファイル AppImageConfig を使用して を作成します。

aws sagemaker create-app-image-config \
    --cli-input-json file://app-image-config-input.json

レスポンスは次のようになります。

{
    "AppImageConfigArn": "arn:aws:sagemaker:us-east-2:acct-id:app-image-config/custom-image-config"
}