Skip to content

Update SageMaker AI instructions #644

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
ncclementi merged 2 commits into from
Dec 18, 2025
Merged

Update SageMaker AI instructions #644

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
1337470
Update SageMaker AI instructions
jacobtomlinson Dec 17, 2025
3f7dbbc
Add SageMaker AI branding
jacobtomlinson Dec 17, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
22 changes: 7 additions & 15 deletions source/cloud/aws/sagemaker.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@ review_priority: "p0"

RAPIDS can be used in a few ways with [AWS SageMaker](https://aws.amazon.com/sagemaker/).

## SageMaker Notebooks
## SageMaker AI Notebooks

To get started head to [the SageMaker console](https://console.aws.amazon.com/sagemaker/) and create a [new SageMaker Notebook Instance](https://console.aws.amazon.com/sagemaker/home#/notebook-instances/create).
To get started head to [the SageMaker AI console](https://console.aws.amazon.com/sagemaker/) and create a [new SageMaker Notebook Instance](https://console.aws.amazon.com/sagemaker/home#/notebook-instances/create).

Choose `Applications and IDEs > Notebooks > Create notebook instance`.

Expand All @@ -17,8 +17,8 @@ Choose `Applications and IDEs > Notebooks > Create notebook instance`.
If a field is not mentioned below, leave the default values:

- **Notebook instance name** = Name of the notebook instance
- **Notebook instance type** = Type of notebook instance. Select a RAPIDS-compatible GPU ([see the RAPIDS docs](https://docs.rapids.ai/install#system-req)) as the SageMaker Notebook instance type (e.g., `ml.p3.2xlarge`).
- **Platform identifier** = 'Amazon Linux 2, Jupyter Lab 4'
- **Notebook instance type** = Type of notebook instance. Select a RAPIDS-compatible GPU ([see the RAPIDS docs](https://docs.rapids.ai/install#system-req)) as the SageMaker Notebook instance type (e.g., `ml.g6.xlarge`).
- **Platform identifier** = 'Amazon Linux 2023, Jupyter Lab 4'

![Screenshot of the create new notebook screen with a ml.p3.2xlarge selected](../../images/sagemaker-create-notebook-instance.png)

Expand All @@ -41,24 +41,16 @@ set -e

sudo -u ec2-user -i <<'EOF'

mamba create -y -n rapids -c rapidsai -c conda-forge -c nvidia rapids=24.12 python=3.12 cuda-version=12.4 \
boto3 \
ipykernel \
'sagemaker-python-sdk>=2.239.0'
eval "$(conda shell.bash hook)"

conda create -y -n rapids {{ rapids_conda_channels }} {{ rapids_sagemaker_conda_packages }} boto3 ipykernel sagemaker-python-sdk
conda activate rapids

python -m ipykernel install --user --name rapids
echo "kernel install completed"
EOF
```

```{warning}
RAPIDS `>24.12` will not be installable on SageMaker Notebook Instances until those instances support
Amazon Linux 2023 or other Linux distributions with GLIBC of at least 2.28.
For more details, see [rapidsai/deployment#520](https://github.com/rapidsai/deployment/issues/520).
```

Set the volume size to at least `15GB`, to accommodate the conda environment.

Then launch the instance.
Expand All @@ -79,7 +71,7 @@ Then in Jupyter select the `rapids` kernel when working with a new notebook.

Once inside JupyterLab you should be able to upload the [Running RAPIDS hyperparameter experiments at scale](/examples/rapids-sagemaker-higgs/notebook) example notebook and continue following those instructions.

## SageMaker Estimators
## SageMaker AI Estimators

RAPIDS can also be used in [SageMaker Estimators](https://sagemaker.readthedocs.io/en/stable/api/training/estimators.html).
Estimators allow you to launch training jobs on ephemeral VMs which SageMaker manages for you.
Expand Down
8 changes: 4 additions & 4 deletions source/conf.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,27 +31,27 @@
"rapids_container": f"nvcr.io/nvidia/rapidsai/base:{stable_version}-cuda12-py3.13",
"rapids_notebooks_container": f"nvcr.io/nvidia/rapidsai/notebooks:{stable_version}-cuda12-py3.13",
"rapids_conda_channel": "rapidsai",
"rapids_conda_channels": "-c rapidsai -c conda-forge -c nvidia",
"rapids_conda_channels": "-c rapidsai -c conda-forge",
"rapids_conda_packages": f"rapids={stable_version} python=3.13 'cuda-version>=12.0,<=12.9'",
"rapids_pip_index": "https://pypi.nvidia.com",
"rapids_pip_version": stable_version,
# SageMaker Notebook Instance examples need to stay pinned to an older RAPIDS until this is resolved:
# https://github.com/rapidsai/deployment/issues/520
"rapids_sagemaker_conda_packages": "rapids=24.12 python=3.12 cuda-version=12.5",
"rapids_sagemaker_conda_packages": f"rapids={stable_version} python=3.12 cuda-version=13",
},
"nightly": {
"rapids_version": f"{nightly_version}",
"rapids_api_docs_version": "nightly",
"rapids_container": f"rapidsai/base:{nightly_version + 'a'}-cuda12-py3.13",
"rapids_notebooks_container": f"rapidsai/notebooks:{nightly_version + 'a'}-cuda12-py3.13",
"rapids_conda_channel": "rapidsai-nightly",
"rapids_conda_channels": "-c rapidsai-nightly -c conda-forge -c nvidia",
"rapids_conda_channels": "-c rapidsai-nightly -c conda-forge",
"rapids_conda_packages": f"rapids={nightly_version} python=3.13 'cuda-version>=12.0,<=12.9'",
"rapids_pip_index": "https://pypi.anaconda.org/rapidsai-wheels-nightly/simple",
"rapids_pip_version": f"{nightly_version}.*,>=0.0.0a0",
# SageMaker Notebook Instance examples need to stay pinned to an older RAPIDS until this is resolved:
# https://github.com/rapidsai/deployment/issues/520
"rapids_sagemaker_conda_packages": "rapids=24.12 python=3.12 cuda-version=12.5",
"rapids_sagemaker_conda_packages": f"rapids={nightly_version} python=3.12 cuda-version=13",
},
}
rapids_version = (
Expand Down