支持的框架映像、Amazon Web Services 区域 和实例类型 - 亚马逊 SageMaker AI
预安装了 SageMaker 探查器的 SageMaker AI 框架映像SageMaker Profiler Python 软件包二进制文件支持 Amazon Web Services 区域支持的实例类型
Amazon Web Services 文档中描述的 Amazon Web Services 服务或功能可能因区域而异。要查看适用于中国区域的差异，请参阅 中国的 Amazon Web Services 服务入门 (PDF)。

本文属于机器翻译版本。若本译文内容与英语原文存在差异，则一律以英文原文为准。

支持的框架映像、Amazon Web Services 区域 和实例类型

此功能支持以下机器学习框架和 Amazon Web Services 区域。

注意

要使用此功能，请确保已安装 SageMaker Python SDK 版本 2.180.0 或更高版本。

预安装了 SageMaker 探查器的 SageMaker AI 框架映像

SageMaker 探查器预安装在以下 Amazon Deep Learning Containers for SageMaker AI 中。

PyTorch 映像

PyTorch 版本 Amazon DLC 映像 URI
2.2.0

763104351884.dkr.ecr.<区域>.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.1.0

763104351884.dkr.ecr.<区域>.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.0.1

763104351884.dkr.ecr.<区域>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker

763104351884.dkr.ecr.<区域>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu121-ubuntu20.04-sagemaker
1.13.1

763104351884.dkr.ecr.<区域>.amazonaws.com/pytorch-training:1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker

TensorFlow 映像

TensorFlow 版本 Amazon DLC 映像 URI
2.13.0

763104351884.dkr.ecr.<区域>.amazonaws.com/tensorflow-training:2.13.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.12.0

763104351884.dkr.ecr.<区域>.amazonaws.com/tensorflow-training:2.12.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.11.0

763104351884.dkr.ecr.<区域>.amazonaws.com/tensorflow-training:2.11.0-gpu-py39-cu112-ubuntu20.04-sagemaker
重要

上表中的框架容器的分发和维护属于 Amazon 深度学习容器服务管理的框架支持政策。如果您使用的先前框架版本不再受支持，我们强烈建议您升级到当前支持的框架版本。

注意

如果您想在其他框架映像或您自己的 Docker 映像中使用 SageMaker Profiler，您可以使用下一节提供的 SageMaker Profiler Python 软件包二进制文件安装 SageMaker Profiler。

SageMaker Profiler Python 软件包二进制文件

如果您想配置自己的 Docker 容器，在其他预构建的 PyTorch 和 TensorFlow 容器中使用 SageMaker Profiler，或在本地安装 SageMaker Profiler Python 软件包，请使用以下二进制文件之一。根据您所处环境中的 Python 和 CUDA 版本，选择以下选项之一。

PyTorch

TensorFlow

有关如何使用二进制文件安装 SageMaker Profiler 的更多信息，请参阅 （可选）安装 SageMaker Profiler Python 软件包。

支持 Amazon Web Services 区域

SageMaker 探查器在以下 Amazon Web Services 区域提供。

  • 美国东部（弗吉尼亚州北部）(us-east-1)

  • 美国东部（俄亥俄州）(us-east-2)

  • 美国西部（俄勒冈州）(us-west-2)

  • 欧洲地区（法兰克福）(eu-central-1)

  • 欧洲地区（爱尔兰）(eu-west-1)

支持的实例类型

SageMaker Profiler 支持对以下实例类型的训练作业进行性能分析。

CPU 和 GPU 性能分析

  • ml.g4dn.12xlarge

  • ml.g5.24xlarge

  • ml.g5.48xlarge

  • ml.p3dn.24xlarge

  • ml.p4de.24xlarge

  • ml.p4d.24xlarge

  • ml.p5.48xlarge

仅限 GPU 性能分析

  • ml.g5.2xlarge

  • ml.g5.4xlarge

  • ml.g5.8xlarge

  • ml.g5.16.xlarge