SageMaker Studioの使っていないKernelを自動でシャットダウンするsagemaker-studio-auto-shutdown-extension

machinelearningaws

SageMaker Studioを使っているとインスタンスを明示的に立ち上げることがないので、シャットダウンするのを忘れて 無駄なインスタンスコストを発生させ続けてしまうことがある。

起動中のKernelおよびインスタンス

Kernelをシャットダウンすると不要になったインスタンスもシャットダウンされるので、使っていないKernelが自動で削除されるようにしたい。 かつてはこれを実現するための仕組みを自前で用意する必要があったが、現在はsagemaker-studio-auto-shutdown-extensionが公式から提供されている。

$ git clone https://github.com/aws-samples/sagemaker-studio-auto-shutdown-extension.git
$ cd sagemaker-studio-auto-shutdown-extension
$ ./install_tarball.sh
$ jupyter serverextension list
...
sagemaker_studio_autoshutdown  enabled 
    - Validating...
      sagemaker_studio_autoshutdown 0.1.0 OK

last activityからの時間が設定値を超過するJupyter Notebook Server APIを呼んでKernelをシャッドダウンするようになっている。

設定

参考

Save costs by automatically shutting down idle resources within Amazon SageMaker Studio | AWS Machine Learning Blog