SageMaker 預先安裝 Profiler 的 AI SageMaker 架構映像 SageMaker Profiler Python 套件二進位檔案支援的 AWS 區域支援的執行個體類型

支援的架構映像 AWS 區域和執行個體類型

此功能支援下列機器學習架構和 AWS 區域。

注意

若要使用此功能，請確定您已安裝 SageMaker Python 2SDK.180.0 版或更新版本。 https://pypi.org/project/sagemaker/2.180.0/

SageMaker 預先安裝 Profiler 的 AI SageMaker 架構映像

SageMaker Profiler 已預先安裝在下列適用於 AWS SageMaker AI 的深度學習容器中。

PyTorch 映像

PyTorch 版本	AWS DLC 映像 URI
2.2.0	`763104351884`.dkr.ecr`<region>`.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.1.0	`763104351884`.dkr.ecr`<region>`.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.0.1	`763104351884`.dkr.ecr`<region>`.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker `763104351884`.dkr.ecr`<region>`.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu121-ubuntu20.04-sagemaker
1.13.1	`763104351884`.dkr.ecr`<region>`.amazonaws.com/pytorch-training:1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker

TensorFlow 映像

TensorFlow 版本	AWS DLC 映像 URI
2.13.0	`763104351884`.dkr.ecr`<region>`.amazonaws.com/tensorflow-training:2.13.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.12.0	`763104351884`.dkr.ecr`<region>`.amazonaws.com/tensorflow-training:2.12.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.11.0	`763104351884`.dkr.ecr`<region>`.amazonaws.com/tensorflow-training:2.11.0-gpu-py39-cu112-ubuntu20.04-sagemaker

重要

上述資料表中架構容器的分佈和維護，是根據 AWS 深度學習容器服務所管理的架構支援政策。如果您使用的是已不再支援的架構版本，強烈建議您升級至目前支援的架構版本。

注意

如果您想要將 SageMaker Profiler 用於其他架構映像或您自己的 Docker 映像，您可以使用下一節提供的 SageMaker Profiler Python 套件二進位檔案來安裝 SageMaker Profiler。

SageMaker Profiler Python 套件二進位檔案

如果您想要設定自己的 Docker 容器，請在 PyTorch 和的其他預先建置容器中使用 SageMaker Profiler TensorFlow，或在本機安裝 SageMaker Profiler Python 套件，請使用下列其中一個二進位檔案。根據環境中的 Python 和CUDA版本，選擇下列其中一項。