View a markdown version of this page

Imágenes de marcos compatibles, Regiones de AWS, y tipos de instancias - Amazon SageMaker AI
SageMaker Imágenes de AI Framework preinstaladas con Profiler SageMakerSageMaker Archivos binarios del paquete Profiler Pythoncompatible Regiones de AWSTipos de instancias admitidas

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

Imágenes de marcos compatibles, Regiones de AWS, y tipos de instancias

Esta característica es compatible con los siguientes marcos de machine learning y Regiones de AWS.

nota

Para usar esta función, asegúrese de haber instalado la versión 2.180.0 o posterior del SDK de SageMaker Python.

SageMaker Imágenes de AI Framework preinstaladas con Profiler SageMaker

SageMaker Profiler viene preinstalado en los siguientes AWS Deep Learning Containers for SageMaker AI.

PyTorch imágenes

PyTorch versiones AWS URI de imagen DLC
2.2.0

763104351884.dkr.ecr. <region>.amazon aws. com/pytorch- formación: 2.2.0-gpu-py310-cu121-ubuntu 20.04-sagemaker
2.1.0

763104351884.dkr.ecr. <region>.amazon aws. com/pytorch- formación: 2.1.0-gpu-py310-cu121-ubuntu 20.04-sagemaker
2.0.1

763104351884.dkr.ecr. <region>.amazon aws. com/pytorch- formación: 2.0.1-gpu-py310-cu118-ubuntu 20.04-sagemaker

763104351884.dkr.ecr. <region>.amazon aws. com/pytorch- formación: 2.0.1-gpu-py310-cu121-ubuntu 20.04-sagemaker
1.13.1

763104351884.dkr.ecr. <region>.amazon aws. com/pytorch- formación: 1.13.1-gpu-py39-cu117-ubuntu 20.04-sagemaker

TensorFlow imágenes

TensorFlow versiones AWS URI de imagen DLC
2.13.0

763104351884.dkr.ecr. <region>.amazon aws. com/tensorflow- formación: 2.13.0-gpu-py310-cu118-ubuntu 20.04-sagemaker
2.12.0

763104351884.dkr.ecr. <region>.amazon aws. com/tensorflow- formación: 2.12.0-gpu-py310-cu118-ubuntu 20.04-sagemaker
2.11.0

763104351884.dkr.ecr. <region>.amazon aws. com/tensorflow- formación: 2.11.0-gpu-py39-cu112-ubuntu 20.04-sagemaker
importante

La distribución y el mantenimiento de los contenedores del marco de las tablas anteriores se rigen por la Política de soporte del marco gestionada por el servicio AWS Deep Learning Containers. Le recomendamos actualizar a las versiones de marco admitidas actualmente si utiliza versiones de marco anteriores que ya no se admiten.

nota

Si desea utilizar SageMaker Profiler para otras imágenes de framework o para sus propias imágenes de Docker, puede instalar SageMaker Profiler utilizando los archivos binarios del paquete Profiler SageMaker Python que se proporcionan en la siguiente sección.

SageMaker Archivos binarios del paquete Profiler Python

Si desea configurar su propio contenedor de Docker, usar SageMaker Profiler en otros contenedores prediseñados o instalar el paquete Profiler SageMaker Python de forma local TensorFlow, use uno de los siguientes archivos binarios. PyTorch En función de las versiones de Python y CUDA de su entorno, elija una de las siguientes opciones.

PyTorch

TensorFlow

Para obtener más información acerca de cómo instalar SageMaker Profiler mediante los archivos binarios, consulte. (Opcional) Instale el paquete SageMaker Profiler Python

compatible Regiones de AWS

SageMaker Profiler está disponible en las siguientes versiones. Regiones de AWS

  • Este de EE. UU. (Norte de Virginia) (us-east-1)

  • Este de EE. UU. (Ohio) (us-east-2)

  • Oeste de EE. UU. (Oregón) (us-west-2)

  • Europa (Fráncfort) (eu-central-1)

  • Europa (Irlanda) (eu-west-1)

Tipos de instancias admitidas

SageMaker Profiler admite la creación de perfiles de los trabajos de formación en los siguientes tipos de instancias.

Generación de perfiles de CPU y GPU

  • ml.g4dn.12xlarge

  • ml.g5.24xlarge

  • ml.g5.48xlarge

  • ml.p3dn.24xlarge

  • ml.p4de.24xlarge

  • ml.p4d.24xlarge

  • ml.p5.48xlarge

Solo generación de perfiles de GPU

  • ml.g5.2xlarge

  • ml.g5.4xlarge

  • ml.g5.8xlarge

  • ml.g5.16.xlarge