Deep Learning Containers 映像 - AWS 深度学习容器
AWS 文档中描述的 AWS 服务或功能可能因区域而异。要查看适用于中国区域的差异,请参阅中国的 AWS 服务入门

Deep Learning Containers 映像

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

Deep Learning Containers Docker 映像在以下区域提供:

区域

代码

通用容器

Elastic Inference 容器

美国东部 (弗吉尼亚北部)

us-east-1

Available

Available

美国东部 (俄亥俄)

us-east-2

Available

Available

美国西部 (加利福尼亚北部)

us-west-1

Available

美国西部 (俄勒冈)

us-west-2

Available

Available

亚太地区 (东京)

ap-northeast-1

Available

Available

亚太地区 (首尔)

ap-northeast-2

Available

Available

亚太地区(香港)

ap-east-1

Available

亚太地区 (孟买)

ap-south-1

Available

亚太地区 (新加坡)

ap-southeast-1

Available

亚太地区 (悉尼)

ap-southeast-2

Available

中东(巴林)

me-south-1

Available

南美洲(圣保罗)

sa-east-1

Available

加拿大 (中部)

ca-central-1

Available

欧洲 (法兰克福)

eu-central-1

Available

欧洲(斯德哥尔摩)

eu-north-1

Available

欧洲 (爱尔兰)

eu-west-1

Available

Available

欧洲 (伦敦)

eu-west-2

Available

欧洲 (巴黎)

eu-west-3

Available

注意:eu-north-1 从 2.0 发行版本开始可用。因此,MXNet-1.4.0 在此区域中不可用。

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

763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:1.15.2-cpu-py27-ubuntu18.04
重要

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

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

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

docker pull <name of container image>

通用框架容器

要使用下表,请选择您所需的框架,以及您正在开始的作业类型和所需的 Python 版本。您的作业类型为 traininginference。您的 Python 版本为 py27py36。如示例 URL 所示,将此信息插入 URL 的可替换部分。

注意

如果您使用 ap-east-1 区域,请将账户 ID“763104351884”替换为“871362719292”。如果您使用 me-south-1 区域,请将账户 ID“763104351884”替换为“217643126080”。

Framework

作业类型

Horovod 选项

CPU/GPU

Python 版本选项

示例 URL

TensorFlow 2.1

训练、推理

训练:是,推理:否

CPU

2.7 (py27),3.6 (py36)

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

TensorFlow 2.1

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.1.0-gpu-py27-cu101-ubuntu18.04

TensorFlow 2.0.1

训练、推理

训练:是,推理:否

CPU

2.7 (py27),3.6 (py36)

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

TensorFlow 2.0.1

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.0.1-gpu-py27-cu100-ubuntu18.04

TensorFlow 1.15.2

训练、推理

训练:是,推理:否

CPU

2.7 (py27),3.6 (py36)

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

TensorFlow 1.15.2

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.2-gpu-py27-cu100-ubuntu18.04

MXNet 1.6.0

训练、推理

训练:是,推理:否

CPU

2.7 (py27),3.6 (py36)

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

MXNet 1.6.0

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

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

训练、推理

CPU

2.7 (py27),3.6 (py36)

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

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

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

Elastic Inference 容器

Framework

作业类型

Horovod 选项

CPU/GPU

Python 版本选项

示例 URL

带弹性推理的 TensorFlow 1.14.0

推理

CPU

2.7 (py27),3.6 (py36)

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

带弹性推理的 MXNet 1.4.1

推理

CPU

2.7 (py27),3.6 (py36)

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

先前版本

Framework

作业类型

Horovod 选项

CPU/GPU

Python 版本选项

示例 URL

TensorFlow 2.0.0

训练、推理

训练:是,推理:否

CPU

2.7 (py27),3.6 (py36)

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

TensorFlow 2.0.0

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.0.0-gpu-py27-cu100-ubuntu18.04

TensorFlow 1.15.0

训练、推理 训练:是,推理:否

CPU

2.7 (py27),3.6 (py36)

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

TensorFlow 1.15.0

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:1.15.0-gpu-py27-cu100-ubuntu18.04

TensorFlow 1.14.0

训练、推理

训练:是,推理:否

CPU

2.7 (py27),3.6 (py36)

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

TensorFlow 1.14.0

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

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

MXNet 1.4.1

训练、推理

CPU

2.7 (py27),3.6 (py36)

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

MXNet 1.4.1

训练、推理

GPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/mxnet-training:1.4.1-gpu-py27-cu100-ubuntu16.04
PyTorch 1.3.1

训练、推理

CPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.3.1-cpu-py27-ubuntu16.04
PyTorch 1.3.1

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.3.1-gpu-py27-cu101-ubuntu16.04
PyTorch 1.2.0

训练、推理

CPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.2.0-cpu-py27-ubuntu16.04
PyTorch 1.2.0

训练、推理

训练:是,推理:否

GPU

2.7 (py27),3.6 (py36)

763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.2.0-gpu-py27-cu100-ubuntu16.04