SageMaker Notebook Instanceの自動停止

概要

SageMaker Notebook Instance を1時間使わないと自動停止するようなスクリプトが github で公開されていたので試します。以下の auto-stop-idle というところです。

Lifecycle config 設定

左のライフサイクル設定のところから設定を作成します。

ノートブックを開始するときはいつでも実行しておきたいので、ノートブックの開始用にスクリプトを設定します。作成画面になったら、エディタのところに on-start.sh の中身を貼りましょう。同じフォルダにあるautostop.pyが、SageMaker の API を使ってノートブックインスタンスを停止するわけですが、これは実行時にダウンロードしてくれます。

IDLE_TIME を変えると、停止までの時間を変えられます。

ノートブックインスタンス作成

作成した LifeCycle Config を指定してノートブックインスタンスを作成しましょう。

作成して待ちます

インスタンスの停止が始まりました

インスタンス起動時間もあるので、1時間4分後にStopping の状態になりました。