Amazon ECS task definitions for deep learning instances
To use deep learning workloads on Amazon ECS, register Amazon EC2 DL1
Considerations
Before you begin deploying DL1 on Amazon ECS, consider the following:
-
Your clusters can contain a mix of DL1 and non-DL1 instances.
-
When creating a service or running a standalone task, you can use instance type attributes specifically when you configure task placement constraints to ensure that your task is launched on the container instance that you specify. Doing so ensures that your resources are used effectively and that your tasks for deep learning workloads are on your DL1 instances. For more information, see How Amazon ECS places tasks on container instances.
The following example runs a task on a
dl1.24xlarge
instance on yourdefault
cluster.aws ecs run-task \ --cluster default \ --task-definition ecs-dl1-task-def \ --placement-constraints type=memberOf,expression="attribute:ecs.instance-type == dl1.24xlarge"
Using a DL1 AMI
You have three options for running an AMI on Amazon EC2 DL1 instances for Amazon ECS:
-
AWS Marketplace AMIs that are provided by Habana here
. -
Habana Deep Learning AMIs that are provided by Amazon Web Services. Because it's not included, you need to install the Amazon ECS container agent separately.
-
Use Packer to build a custom AMI that's provided by the GitHub repo
. For more information, see the Packer documentation .