支持的框架映像和实例类型 Amazon Web Services 区域 - 亚马逊 SageMaker AI
SageMaker Profiler 中预装的 AI 框架镜像 SageMaker 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 Profiler 中预装的 AI 框架镜像 SageMaker

SageMaker Profiler 已预装在以下适用 SageMaker 于 AI 的 Dee Amazon p Learning Contain ers 中。

PyTorch图片

PyTorch 版本 Amazon 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 版本 Amazon 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
重要

上表中框架容器的分发和维护受到 Dee Amazon p Learning Containers 服务管理的框架支持政策的约束。如果您使用的先前框架版本不再受支持，我们强烈建议您升级到当前支持的框架版本。

注意

如果你想将 P SageMaker rofiler 用于其他框架镜像或你自己的 Docker 镜像，你可以使用下一节中提供的 P SageMaker rofiler SageMaker Python 包二进制文件来安装 Profiler。

SageMaker Profiler Python 包二进制文件

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

PyTorch

TensorFlow

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

支持的 Amazon Web Services 区域

SageMaker Profiler 在以下 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