支持的框架映像、AWS 区域 和实例类型 - Amazon SageMaker
预安装 SageMaker Profiler 的 SageMaker 框架映像SageMaker Profiler Python 软件包二进制文件支持 AWS 区域支持的实例类型

支持的框架映像、AWS 区域 和实例类型

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

注意

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

预安装 SageMaker Profiler 的 SageMaker 框架映像

SageMaker Profiler 预安装在以下适用于 SageMaker 的 AWS 深度学习容器中。

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 深度学习容器服务管理的框架支持政策。如果您使用的先前框架版本不再受支持，我们强烈建议您升级到当前支持的框架版本。

注意

如果您想在其他框架映像或您自己的 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 软件包。

支持 AWS 区域

SageMaker 探查器在以下 AWS 区域提供。

  • 美国东部（弗吉尼亚州北部）(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