The following table lists the Docker image URLs that will be used by Amazon ECS in task definitions. Replace the <repository-name> and <image-tag> values based on your desired container.
Once you've selected your desired Deep Learning Containers image, continue with the one of the following:
-
To run training and inference on Deep Learning Containers for Amazon EC2 using MXNet, PyTorch, TensorFlow, and TensorFlow 2, see Amazon EC2 Tutorials
-
To run training and inference on Deep Learning Containers for Amazon ECS using MXNet, PyTorch, and TensorFlow, see Amazon ECS tutorials
-
Deep Learning Containers for Amazon EKS offer CPU, GPU, and distributed GPU-based training, as well as CPU and GPU-based inference. To run training and inference on Deep Learning Containers for Amazon EKS using MXNet, PyTorch, and TensorFlow, see Amazon EKS Tutorials
-
For information on security in Deep Learning Containers, see Security in AWS Deep Learning Containers
-
For a list of the latest Deep Learning Containers release notes, see Release Notes for Deep Learning Containers
Deep Learning Containers Docker Images are available in the following regions:
| Region | Code | General Container | Elastic Inference Container | Neuron Container | Example URL |
|---|---|---|---|---|---|
| US East (N. Virginia) | us-east-1 | Available | Available | Available | 763104351884.dkr.ecr.us-east-1.amazonaws.com/<repository-name>:<image-tag> |
| US East (Ohio) | us-east-2 | Available | Available | Available | 763104351884.dkr.ecr.us-east-2.amazonaws.com/<repository-name>:<image-tag> |
| US West (N. California) | us-west-1 | Available | None | None | 763104351884.dkr.ecr.us-west-1.amazonaws.com/<repository-name>:<image-tag> |
| US West (Oregon) | us-west-2 | Available | Available | Available | 763104351884.dkr.ecr.us-west-2.amazonaws.com/<repository-name>:<image-tag> |
| Africa (Cape Town) | af-south-1 | Available | Available | None | 626614931356.dkr.ecr.af-south-1.amazonaws.com/<repository-name>:<image-tag> |
| Asia Pacific (Hong Kong) | ap-east-1 | Available | None | None | 871362719292.dkr.ecr.ap-east-1.amazonaws.com/<repository-name>:<image-tag> |
| Asia Pacific (Mumbai) | ap-south-1 | Available | None | Available | 763104351884.dkr.ecr.ap-south-1.amazonaws.com/<repository-name>:<image-tag> |
| Asia Pacific (Seoul) | ap-northeast-2 | Available | Available | None | 763104351884.dkr.ecr.ap-northeast-2.amazonaws.com/<repository-name>:<image-tag> |
| Asia Pacific (Singapore) | ap-southeast-1 | Available | None | Available | 763104351884.dkr.ecr.ap-southeast-1.amazonaws.com/<repository-name>:<image-tag> |
| Asia Pacific (Sydney) | ap-southeast-2 | Available | None | Available | 763104351884.dkr.ecr.ap-southeast-2.amazonaws.com/<repository-name>:<image-tag> |
| Asia Pacific (Tokyo) | ap-northeast-1 | Available | Available | Available | 763104351884.dkr.ecr.ap-northeast-1.amazonaws.com/<repository-name>:<image-tag> |
| Canada (Central) | ca-central-1 | Available | None | None | 763104351884.dkr.ecr.ca-central-1.amazonaws.com/<repository-name>:<image-tag> |
| EU (Frankfurt) | eu-central-1 | Available | None | Available | 763104351884.dkr.ecr.eu-central-1.amazonaws.com/<repository-name>:<image-tag> |
| EU (Ireland) | eu-west-1 | Available | Available | Available | 763104351884.dkr.ecr.eu-west-1.amazonaws.com/<repository-name>:<image-tag> |
| EU (London) | eu-west-2 | Available | None | None | 763104351884.dkr.ecr.eu-west-2.amazonaws.com/<repository-name>:<image-tag> |
| EU (Milan) | eu-south-1 | Available | None | None | 692866216735.dkr.ecr.eu-south-1.amazonaws.com/<repository-name>:<image-tag> |
| EU (Paris) | eu-west-3 | Available | None | Available | 763104351884.dkr.ecr.eu-west-3.amazonaws.com/<repository-name>:<image-tag> |
| EU (Stockholm) | eu-north-1 | Available | None | None | 763104351884.dkr.ecr.eu-north-1.amazonaws.com/<repository-name>:<image-tag> |
| Middle East (Bahrain) | me-south-1 | Available | None | None | 217643126080.dkr.ecr.me-south-1.amazonaws.com/<repository-name>:<image-tag> |
| South America (Sao Paulo) | sa-east-1 | Available | None | Available | 763104351884.dkr.ecr.sa-east-1.amazonaws.com/<repository-name>:<image-tag> |
| China (Beijing) | cn-north-1 | Available | None | None | 727897471807.dkr.ecr.cn-north-1.amazonaws.com.cn/<repository-name>:<image-tag> |
| China (Ningxia) | cn-northwest-1 | Available | None | None | 727897471807.dkr.ecr.cn-northwest-1.amazonaws.com.cn/<repository-name>:<image-tag> |
Note: eu-north-1 is available starting in the version 2.0 release. As a result, MXNet-1.4.0 is not available in this region.
ECR is a regional service and the Image table contains the URLs for us-east-1 images. To pull from one of the regions mentioned previously, insert the region in the repository URL following this example:
763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:1.15.2-cpu-py27-ubuntu18.04
Important
You must login to access to the DLC image repository before pulling the image. Ensure your CLI is up to date using the steps in Installing the current AWS CLI Version Then, specify your region and its corresponding ECR Registry from the previous table in the following command:
aws ecr get-login-password --region us-east-1 | docker login --username AWS --password-stdin 763104351884.dkr.ecr.us-east-1.amazonaws.com
You can then pull these Docker images from ECR by running:
docker pull <name of container image>
To use the following table, select your desired framework, as well as
the kind of job you're starting, and the desired Python version. Your
job type is either training or inference. Your Python version is
either py27, py36, or py37 depending on availability. Plug this information into the replaceable
portions of the URL as shown in the example URL.
You can pin your version by adding the version tag to your URL as follows:
763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.2.0-cpu-py37-ubuntu18.04-v1.0
| Framework | Job Type | Horovod Options | CPU/GPU | Python Version Options | Example URL | Latest Version |
|---|---|---|---|---|---|---|
| TensorFlow 2.3.1 | training | Yes | CPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.3.1-cpu-py37-ubuntu18.04 | v1.12 |
| TensorFlow 2.3.1 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.3.1-gpu-py37-cu110-ubuntu18.04 | v3.2 |
| TensorFlow 2.3.1 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.3.1-gpu-py37-cu102-ubuntu18.04 | v2.9 |
| TensorFlow 2.3.1 | inference | No | CPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.3.1-cpu-py37-ubuntu18.04 | v1.7 |
| TensorFlow 2.3.1 | inference | No | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.3.1-gpu-py37-cu102-ubuntu18.04 | v1.7 |
| TensorFlow 1.15.4 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.4-gpu-py37-cu110-ubuntu18.04 | v8.1 |
| TensorFlow 1.15.4 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.4-gpu-py37-cu100-ubuntu18.04 | v7.5 |
| TensorFlow 1.15.4 | training | Yes | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.4-gpu-py36-cu100-ubuntu18.04 | v4.9 |
| TensorFlow 1.15.4 | inference | No | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.15.4-gpu-py36-cu100-ubuntu18.04 | v4.8 |
| TensorFlow 1.15.4 | training | Yes | CPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.4-cpu-py37-ubuntu18.04 | v7.7 |
| TensorFlow 1.15.4 | training | Yes | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.4-cpu-py36-ubuntu18.04 | v4.9 |
| TensorFlow 1.15.4 | inference | No | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.15.4-cpu-py36-ubuntu18.04 | v4.8 |
| MXNet 1.8.0 | training | Yes | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.8.0-gpu-py37-cu110-ubuntu16.04 | v1.0 |
| MXNet 1.8.0 | training | Yes | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.8.0-cpu-py37-ubuntu16.04 | v1.0 |
| MXNet 1.8.0 | inference | No | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference:1.8.0-gpu-py37-cu110-ubuntu16.04 | v1.0 |
| MXNet 1.8.0 | inference | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference:1.8.0-cpu-py37-ubuntu16.04 | v1.0 |
| PyTorch 1.6.0 | training | Yes | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.6.0-gpu-py36-cu110-ubuntu18.04 | v3.0 |
| PyTorch 1.6.0 | training | Yes | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.6.0-gpu-py36-cu110-ubuntu16.04 | v2.2 |
| PyTorch 1.6.0 | training | Yes | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.6.0-gpu-py36-cu101-ubuntu16.04 | v1.5 |
| PyTorch 1.6.0 | training | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.6.0-cpu-py36-ubuntu16.04 | v1.6 |
| PyTorch 1.6.0 | inference | No | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:1.6.0-gpu-py36-cu101-ubuntu16.04 | v1.6 |
| PyTorch 1.6.0 | inference | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:1.6.0-cpu-py36-ubuntu16.04 | v1.6 |
| Framework | Job Type | Horovod Options | CPU/GPU | Python Version Options | Example URL |
|---|---|---|---|---|---|
| TensorFlow 1.15.0 with Elastic Inference | inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-eia:1.15.0-cpu-py36-ubuntu18.04 |
| TensorFlow 2.0.0 with Elastic Inference | inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-eia:2.0.0-cpu-py36-ubuntu18.04 |
| MXNet 1.5.1 with Elastic Inference | inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference-eia:1.5.1-cpu-py36-ubuntu16.04 |
| PyTorch 1.5.1 with Elastic Inference | inference | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-eia:1.5.1-cpu-py36-ubuntu16.04 |
| Framework | Job Type | Python Version Options | Example URL |
|---|---|---|---|
| TensorFlow 1.15 with Neuron Inference | inference | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-neuron:1.15.4-neuron-py37-ubuntu18.04 |
| Framework | Job Type | Horovod Options | CPU/GPU | Python Version Options | Example URL |
|---|---|---|---|---|---|
| TensorFlow 1.15.3 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.3-gpu-py37-cu100-ubuntu18.04 |
| TensorFlow 1.15.3 | training | Yes | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.3-gpu-py36-cu100-ubuntu18.04 |
| TensorFlow 1.15.3 | inference | No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.15.3-gpu-py36-cu100-ubuntu18.04 |
| TensorFlow 1.15.3 | training | Yes | CPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.3-cpu-py37-ubuntu18.04 |
| TensorFlow 1.15.3 | training | Yes | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.3-cpu-py36-ubuntu18.04 |
| TensorFlow 1.15.3 | inference | No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.15.3-cpu-py36-ubuntu18.04 |
| MXNet 1.7.0 | training | Yes | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.7.0-gpu-py36-cu101-ubuntu16.04 |
| MXNet 1.7.0 | training | Yes | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.7.0-cpu-py36-ubuntu16.04 |
| MXNet 1.7.0 | inference | No | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference:1.7.0-gpu-py36-cu101-ubuntu16.04 |
| MXNet 1.7.0 | inference | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference:1.7.0-cpu-py36-ubuntu16.04 |
| TensorFlow 2.2.1 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.2.1-gpu-py37-cu102-ubuntu18.04 |
| TensorFlow 2.2.1 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.2.1-gpu-py37-cu101-ubuntu18.04 |
| TensorFlow 2.2.1 | training | Yes | CPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.2.1-cpu-py37-ubuntu18.04 |
| TensorFlow 2.2.1 | inference | No | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.2.1-gpu-py37-cu102-ubuntu18.04 |
| TensorFlow 2.2.1 | inference | No | CPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.2.1-cpu-py37-ubuntu18.04 |
| TensorFlow 2.2.0 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.2.0-gpu-py37-cu102-ubuntu18.04 |
| TensorFlow 2.2.0 | training | Yes | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.2.0-gpu-py37-cu101-ubuntu18.04 |
| TensorFlow 2.2.0 | training | Yes | CPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.2.0-cpu-py37-ubuntu18.04 |
| TensorFlow 2.2.0 | inference | No | GPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.2.0-gpu-py37-cu102-ubuntu18.04 |
| TensorFlow 2.2.0 | inference | No | CPU | 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:2.2.0-cpu-py37-ubuntu18.04 |
| MXNet 1.6.0 | training | Yes | GPU | 2.7 (py27) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.6.0-gpu-py27-cu101-ubuntu16.04 |
| MXNet 1.6.0 | training, inference | Training: Yes, Inference: No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.6.0-gpu-py36-cu101-ubuntu16.04 |
| MXNet 1.6.0 | training, inference | Training: Yes, Inference: No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.6.0-cpu-py36-ubuntu16.04 |
| PyTorch 1.5.1 | training | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.5.1-cpu-py36-ubuntu16.04 |
| PyTorch 1.5.1 | training | Yes | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.5.1-gpu-py36-cu101-ubuntu16.04 |
| PyTorch 1.5.1 | inference | No | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:1.5.1-gpu-py36-cu101-ubuntu16.04 |
| PyTorch 1.5.1 | inference | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:1.5.1-cpu-py36-ubuntu16.04 |
| TensorFlow 2.1.1 | training, inference | Training: Yes, Inference: No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.1.1-cpu-py36-ubuntu18.04 |
| TensorFlow 2.1.1 | training, inference | Training: Yes, Inference: No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.1.1-gpu-py36-cu101-ubuntu18.04 |
| PyTorch 1.5.0 | training, inference | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.5.0-cpu-py36-ubuntu16.04 |
| PyTorch 1.5.0 | training, inference | Training: Yes, Inference: No | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.5.0-gpu-py36-cu101-ubuntu16.04 |
| TensorFlow 1.15.2 | training | Yes | CPU | 2.7 (py27), 3.6 (py36), 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.2-cpu-py37-ubuntu18.04 |
| TensorFlow 1.15.2 | inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.15.2-cpu-py36-ubuntu18.04 |
| TensorFlow 1.15.2 | training | Yes | GPU | 2.7 (py27), 3.6 (py36), 3.7 (py37) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.2-gpu-py37-cu100-ubuntu18.04 |
| TensorFlow 1.15.2 | inference | No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.15.2-gpu-py36-cu100-ubuntu18.04 |
| MXNet 1.4.1 with Elastic Inference | inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference-eia:1.4.1-cpu-py36-ubuntu16.04 |
| PyTorch 1.4.0 | training | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.4.0-cpu-py36-ubuntu16.04 |
| PyTorch 1.4.0 | inference | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:1.4.0-cpu-py36-ubuntu16.04 |
| PyTorch 1.4.0 | training | Yes | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.4.0-gpu-py36-cu101-ubuntu16.04 |
| PyTorch 1.4.0 | inference | No | GPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:1.4.0-gpu-py36-cu101-ubuntu16.04 |
| TensorFlow 2.0.2 | training, inference | Training:Yes, Inference: No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.0.2-cpu-py36-ubuntu18.04 |
| TensorFlow 2.0.2 | training, inference | Training:Yes, Inference: No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.0.2-gpu-py36-cu100-ubuntu18.04 |
| TensorFlow 1.14.0 | training, inference | Training:Yes, Inference: No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.14.0-cpu-py36-ubuntu16.04 |
| TensorFlow 1.14.0 | training, inference | Training:Yes, Inference: No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.14.0-gpu-py36-cu100-ubuntu16.04 |
| MXNet 1.4.1 | training, inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.4.1-cpu-py36-ubuntu16.04 |
| MXNet 1.4.1 | training, inference | No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.4.1-gpu-py36-cu100-ubuntu16.04 |
| PyTorch 1.3.1 | training, inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.3.1-cpu-py36-ubuntu16.04 |
| PyTorch 1.3.1 | training, inference | Training:Yes, Inference: No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.3.1-gpu-py36-cu101-ubuntu16.04 |
| PyTorch 1.2.0 | training, inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.2.0-cpu-py36-ubuntu16.04 |
| PyTorch 1.2.0 | training, inference | Training:Yes, Inference: No | GPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.2.0-gpu-py36-cu100-ubuntu16.04 |
| Framework | Job Type | Horovod Options | CPU/GPU | Python Version Options | Example URL |
|---|---|---|---|---|---|
| PyTorch 1.3.1 with Elastic Inference | inference | No | CPU | 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference-eia:1.3.1-cpu-py36-ubuntu16.04 |
| TensorFlow 1.14.0 with Elastic Inference | inference | No | CPU | 2.7 (py27), 3.6 (py36) | 763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference-eia:1.14.0-cpu-py36-ubuntu16.04 |