As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.
AWS Batch suporte para trabalhos de treinamento em SageMaker IA
Uma fila de trabalhos do AWS Batch armazena e prioriza os trabalhos enviados antes de serem executados em recursos de computação. Você pode enviar trabalhos de treinamento de SageMaker IA para uma fila de trabalhos para aproveitar as ferramentas de agendamento e priorização de tarefas sem servidor fornecidas pela. AWS Batch
Como funciona
As etapas a seguir descrevem o fluxo de trabalho de como usar uma fila de AWS Batch trabalho com trabalhos de treinamento de SageMaker IA. Para ver tutoriais mais detalhados e exemplos de caderno, consulte a seção Conceitos básicos.
-
Configuração AWS Batch e todas as permissões necessárias. Para obter mais informações, consulte Configurar o AWS Batch no Manual do usuário do AWS Batch .
-
Crie os seguintes AWS Batch recursos no console ou usando o AWS CLI:
-
Ambiente de serviço — contém parâmetros de configuração para integração com a SageMaker IA.
-
SageMaker Fila de vagas de treinamento em IA — Integra-se à SageMaker IA para enviar vagas de treinamento.
-
-
Configure seus detalhes e solicite um trabalho de treinamento de SageMaker IA, como a imagem do contêiner de treinamento. Para enviar um trabalho de treinamento para uma AWS Batch fila, você pode usar o SDK AWS CLI AWS SDK para Python (Boto3), o ou o SageMaker AI Python SDK.
-
Envie suas tarefas de treinamento à fila de trabalhos. É possível usar as seguintes opções para enviar tarefas:
-
Use a API do AWS Batch SubmitServiceJob.
-
Use o
aws_batchmódulodo SDK do SageMaker AI Python. Depois de criar um TrainingQueue objeto e um modelo de objeto de treinamento (como um ModelTrainer), você pode enviar trabalhos de treinamento para o TrainingQueue usando o queue.submit()método.
-
-
Depois de enviar trabalhos, veja a fila de trabalhos e o status do trabalho com o AWS Batch console, a AWS Batch DescribeServiceJob API ou a API SageMaker AI DescribeTrainingJob.
Custo e disponibilidade
Para obter informações detalhadas sobre preços sobre vagas de treinamento, consulte a definição de preço do Amazon SageMaker AI
Você pode usar AWS Batch para trabalhos de treinamento de SageMaker IA em qualquer Região da AWS lugar onde haja vagas de treinamento disponíveis. Para obter mais informações, consulte Endpoints e cotas de SageMaker IA da Amazon.
Para garantir que você tenha a capacidade necessária quando precisar, você pode usar os planos de treinamento flexíveis (FTP) da SageMaker IA. Esses planos permitem que você reserve capacidade para suas tarefas de treinamento. Quando combinado com os recursos AWS Batch de filas, você pode maximizar a utilização durante a duração do seu plano. Para obter mais informações, consulte Reservar planos de treinamento para seus cargos ou HyperPod clusters de treinamento.
Conceitos básicos
Para ver um tutorial sobre como configurar uma fila de AWS Batch vagas e enviar vagas de treinamento em SageMaker IA, consulte Introdução à SageMaker IA AWS Batch no Guia do AWS Batch usuário.
Para notebooks Jupyter que mostram como usar o aws_batch módulo no SDK do SageMaker AI Python, consulte os exemplos de cadernos de tarefas de treinamento em SageMaker IA no repositório AWS Batch amazon-sagemaker-examples. GitHub
