Definições de tarefa do Amazon ECS para instâncias de aprendizado profundo - Amazon Elastic Container Service
ConsideraçõesUtilizar uma AMI DL1

Definições de tarefa do Amazon ECS para instâncias de aprendizado profundo

Para utilizar workloads de aprendizado profundo no Amazon ECS, registre instâncias DL1 do Amazon EC2 nos seus clusters. Instâncias DL1 do Amazon EC2 contam com a tecnologia dos aceleradores Gaudi da Habana Labs (uma empresa Intel). Utilize o Habana SynapseAI SDK para se conectar aos aceleradores Habana Gaudi. O SDK é compatível com frameworks de machine learning populares, como TensorFlow e PyTorch.

Considerações

Antes de começar a implantar a DL1 no Amazon ECS, considere o seguinte:

  • Seus clusters podem conter uma combinação de instâncias DL1 e não DL1.

  • Ao criar um serviço ou executar uma tarefa autônoma, você pode usar atributos de tipo de instância especificamente ao configurar restrições de posicionamento de tarefas para garantir que sua tarefa seja iniciada na instância de contêiner que você especificar. Isso garante que seus recursos sejam usados com eficiência e que suas tarefas para workloads de aprendizado profundo estejam em suas instâncias DL1. Para ter mais informações, consulte Como o Amazon ECS posiciona tarefas em instâncias de contêineres.

    O exemplo a seguir executa uma tarefa em uma instância dl1.24xlarge do cluster default.

    aws ecs run-task \
     --cluster default \
     --task-definition ecs-dl1-task-def \
     --placement-constraints type=memberOf,expression="attribute:ecs.instance-type == dl1.24xlarge"

Utilizar uma AMI DL1

Existem três opções para executar uma AMI em instâncias DL1 do Amazon EC2 para o Amazon ECS:

  • AMIs do AWS Marketplace que são fornecidas pela Habana aqui.

  • AMIs do Habana Deep Learning que são fornecidas pela Amazon Web Services. Como isso não está incluído, é necessário instalar o agente de contêiner do Amazon ECS separadamente.

  • Use o Packer para criar uma AMI personalizada que é fornecida pelo repositório do GitHub. Para obter mais informações, consulte a documentação do Packer.