SageMaker Studioでカスタムイメージが起動しない

発生した事象

SageMaker Studioのjupyter labでカスタムイメージを選択して起動しようとしたところ。ステータスがUpdattingのまま起動されない。(2024/4/1現在）

原因

サポートに問い合わせたところ、AWS側の問題（おそらくバグ？）とのことでした。
space の JupyterLabAppSettings に SageMakerImageVersionAlias が設定されている場合、カスタムイメージを使用してJupyterLab App を起動しようとすると、CreateApp APIが呼び出されません。

spaceの設定確認コマンド[2]

aws sagemaker describe-space --domain-id <Domain ID> --space-name <space 名>

対処法

対処法として以下を提示して頂きました。

1. 新たに作成された space を使用する

2. AWS CLI コマンドから JupyterLab App を起動する[1]

   aws sagemaker create-app \
   --domain-id <ドメイン ID> \
   --app-type JupyterLab \
   --app-name default \
   --space-name <space 名> \
   --resource-spec "SageMakerImageArn=arn:aws:sagemaker:ap-northeast-1:<AWS アカウント ID>:image/<カスタムイメージ名>,SageMakerImageVersionAlias=1,InstanceType=<インスタンスタイプ>"
   

3. JupyterLab の space 一覧ページから JupyterLab App を起動する。
   space の JupyterLabAppSettings における SageMakerImageArn の値がカスタムイメージの ARN になっていれば、JupyterLab の space 一覧ページ から [Run] をクリックすることで起動できる。[2]

参考資料

[1] CreateApp - Amazon SageMaker
https://docs.aws.amazon.com/sagemaker/latest/APIReference/API_CreateApp.html

[2] describe-space — AWS CLI 2.15.34 Command Reference
https://awscli.amazonaws.com/v2/documentation/api/latest/reference/sagemaker/describe-space.html

[3] update-space — AWS CLI 2.15.34 Command Reference
https://awscli.amazonaws.com/v2/documentation/api/latest/reference/sagemaker/update-space.html