LCC スクリプトを使用した SageMaker ノートブックインスタンスのカスタマイズ

ノートブックインスタンスへのパッケージまたはサンプルノートブックをインストールする

ノートブックインスタンスのネットワークとセキュリティを設定する

シェルスクリプトを使用してノートブックインスタンスをカスタマイズする

ライフサイクル設定は、root ユーザーとして実行されます。スクリプトが /home/ec2-user/SageMaker ディレクトリ内で何らかの変更を行う場合 (例えば、pip を使用してパッケージをインストールする場合) は、ec2-user ユーザーとして sudo -u ec2-user コマンドを実行します。これは Amazon SageMaker AI が実行するのと同じユーザーです。

SageMaker AI ノートブックインスタンスは、conda環境を使用して Jupyter Notebook に異なるカーネルを実装します。1 つ以上のノートブックカーネルで使用可能なパッケージをインストールするには、インストールするパッケージのカーネルを含む conda 環境をアクティブにする conda 環境コマンドで、パッケージをインストールするためのコマンドを囲みます。

例えば、python3 環境にのみパッケージをインストールする場合は、次のコードを使用します。

#!/bin/bash
sudo -u ec2-user -i <<EOF

# This will affect only the Jupyter kernel called "conda_python3".
source activate python3

# Replace myPackage with the name of the package you want to install.
pip install myPackage
# You can also perform "conda install" here as well.

source deactivate

EOF

ノートブックインスタンス内のすべての conda 環境にパッケージをインストールする場合は、次のコードを使用します。

#!/bin/bash
sudo -u ec2-user -i <<EOF

# Note that "base" is special environment name, include it there as well.
for env in base /home/ec2-user/anaconda3/envs/*; do
    source /home/ec2-user/anaconda3/bin/activate $(basename "$env")

    # Installing packages in the Jupyter system environment can affect stability of your SageMaker
    # Notebook Instance.  You can remove this check if you'd like to install Jupyter extensions, etc.
    if [ $env = 'JupyterSystemEnv' ]; then
      continue
    fi

    # Replace myPackage with the name of the package you want to install.
    pip install --upgrade --quiet myPackage
    # You can also perform "conda install" here as well.

    source /home/ec2-user/anaconda3/bin/deactivate
done

EOF

すべての conda 環境は、デフォルトの環境フォルダ (/home/user/anaconda3/envs) に保存する必要があります。