深度学习 AMI
开发人员指南
AWS 文档中描述的 AWS 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅中国的 AWS 服务入门

Deep Learning Containers映像

AWS Deep Learning Containers可用作 Amazon ECR 中的 Docker 映像。每个 Docker 映像专用于在带 CPU 或 GPU 支持的特定深度学习框架版本、python 版本上运行训练或推理。下表列出了任务定义中的 Amazon ECS 将使用的 Docker 映像 URL。

AWS Deep Learning Containers Docker 映像在以下区域中可用:us-east-1、us-east-2、us-west-1、us-west-2、ap-northeast-1、ap-northeast-2、ap-south-1、ap-southeast-1、ap-southeast-2、sa-east-1、ca-central-1、eu-central-1、eu-west-1、eu-west-2、eu-west-3

ECR 是一项区域性服务并且映像表包含 us-east-1 映像的 URL。要从之前提到的区域之一拉取映像,请将该区域插入此示例后面的存储库 URL 中:

763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:1.13-cpu-py27-ubuntu16.04

重要

要拉取映像,必须先登录并访问 映像存储库。在以下命令中指定您的区域:

$(aws ecr get-login --no-include-email --region us-east-1 --registry-ids 763104351884)

然后,运行以下命令,从 Amazon ECR 中拉取这些 Docker 映像:

docker pull <name of container image>

框架

训练/推理

Horovod

CPU/GPU

Python 版本

URL

TensorFlow

训练

CPU

2.7

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.13-cpu-py27-ubuntu16.04

TensorFlow

训练

CPU

3.6

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.13-cpu-py36-ubuntu16.04

TensorFlow

训练

GPU

2.7

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.13-horovod-gpu-py27-cu100-ubuntu16.04

TensorFlow

训练

GPU

3.6

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.13-horovod-gpu-py36-cu100-ubuntu16.04

TensorFlow

推理

CPU

2.7

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.13-cpu-py27-ubuntu16.04

TensorFlow

推理

CPU

3.6

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.13-cpu-py36-ubuntu16.04

TensorFlow

推理

GPU

2.7

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.13-gpu-py27-cu100-ubuntu16.04

TensorFlow

推理

GPU

3.6

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-inference:1.13-gpu-py36-cu100-ubuntu16.04

MXNet

训练

CPU

2.7

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.4.0-cpu-py27-ubuntu16.04

MXNet

训练

CPU

3.6

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.4.0-cpu-py36-ubuntu16.04

MXNet

训练

GPU

2.7

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.4.0-gpu-py27-cu90-ubuntu16.04

MXNet

训练

GPU

3.6

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.4.0-gpu-py36-cu90-ubuntu16.04

MXNet

推理

CPU

2.7

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference:1.4.0-cpu-py27-ubuntu16.04

MXNet

推理

CPU

3.6

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference:1.4.0-cpu-py36-ubuntu16.04

MXNet

推理

GPU

2.7

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference:1.4.0-gpu-py27-cu90-ubuntu16.04

MXNet

推理

GPU

3.6

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-inference:1.4.0-gpu-py36-cu90-ubuntu16.04