使自定义内核可用于 SageMaker Studio 笔记本答案

【问题标题】：Make custom kernels available to SageMaker Studio notebooks使自定义内核可用于 SageMaker Studio 笔记本
【发布时间】：2021-09-26 16:04:41
【问题描述】：

在启动 SageMaker Studio 服务器时，我只能看到一组预定义的内核我为任何笔记本选择内核。

我通过将.condarc 指向存储在 EFS 上的自定义 miniconda 目录来创建 conda 环境并在会话之间保持它们。

我希望所有笔记本都可以访问存储在自定义 miniconda 目录中的环境。我可以在系统终端上执行此操作，但似乎找不到使内核可用于笔记本的方法。

我知道Life Cycle Configuration，但这似乎只适用于笔记本实例而不是 SageMaker Studio。

期望的结果

理想情况下，让笔记本电脑可以持久使用自定义内核，但如果这不可行或需要自定义 docker 映像，我很乐意每次运行服务器时手动运行脚本。

到目前为止我所做的尝试：

我运行了以下代码，这是 start.sh 的调整版本，旨在用于生命周期配置。

#!/bin/bash

set -e

sudo -u sagemaker-user -i <<'EOF'
unset SUDO_UID
WORKING_DIR=/home/sagemaker-user/.SageMaker/custom-miniconda/
source "$WORKING_DIR/miniconda/bin/activate"
for env in $WORKING_DIR/miniconda/envs/*; do
    BASENAME=$(basename "$env")
    source activate "$BASENAME"
    python -m ipykernel install --user --name "$BASENAME" --display-name "$BASENAME"
done

EOF

这不起作用，我无法从笔记本访问内核。

【问题讨论】：

标签： amazon-web-services conda amazon-sagemaker

【解决方案1】：

如果您需要 SageMaker Studio 中的持久自定义内核，您可以创建 ECR 存储库并使用自定义环境配置构建 docker 映像。然后可以将此图像附加到 SageMaker 工作室笔记本。 Reference link!

SageMaker Studio 现在还支持使用生命周期配置。 Reference link!

【讨论】：