Skip to content

Commit 9f248c5

Browse files
saimidusaimidu
authored
[doc] Add PT 1.7.1 Inference and TF 2.4.1 Training to Available Images (#898)
* [doc] Add PT 1.7.1 Inference DLCs to Available Images * Update available_images.md
1 parent 2d53c0b commit 9f248c5

File tree

1 file changed

+9
-5
lines changed

1 file changed

+9
-5
lines changed

‎available_images.md

Lines changed: 9 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -86,9 +86,8 @@ You can pin your version by adding the version tag to your URL as follows:
8686

8787
| Framework |Job Type |Horovod Options|CPU/GPU |Python Version Options |Example URL |Latest Version |
8888
|-------------------|-----------|---------------|-----------|-----------------------|---------------------------------------------------------------------------------------------------|---------------|
89-
|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 |
90-
|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 |
91-
|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 |
89+
|TensorFlow 2.4.1 |training |Yes |CPU | 3.7 (py37) |763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.4.1-cpu-py37-ubuntu18.04 |v1.0 |
90+
|TensorFlow 2.4.1 |training |Yes |GPU | 3.7 (py37) |763104351884.dkr.ecr.us-east-1.amazonaws.com/tensorflow-training:2.4.1-gpu-py37-cu110-ubuntu18.04 |v1.0 |
9291
|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 |
9392
|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 |
9493
|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 |
@@ -102,8 +101,10 @@ You can pin your version by adding the version tag to your URL as follows:
102101
|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 |
103102
|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 |
104103
|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 |
105-
|PyTorch 1.7.1 |training |Yes |GPU | 3.6 (py36) |763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.7.1-gpu-py36-cu110-ubuntu18.04 |v1.0 |
106-
|PyTorch 1.7.1 |training |No |CPU | 3.6 (py36) |763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.7.1-cpu-py36-ubuntu18.04 |v1.0 |
104+
|PyTorch 1.7.1 |training |Yes |GPU | 3.6 (py36) |763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.7.1-gpu-py36-cu110-ubuntu18.04 |v1.1 |
105+
|PyTorch 1.7.1 |training |No |CPU | 3.6 (py36) |763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-training:1.7.1-cpu-py36-ubuntu18.04 |v1.1 |
106+
|PyTorch 1.7.1 |inference |No |GPU | 3.6 (py36) |763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:1.7.1-gpu-py36-cu110-ubuntu18.04 |v1.1 |
107+
|PyTorch 1.7.1 |inference |No |CPU | 3.6 (py36) |763104351884.dkr.ecr.us-east-1.amazonaws.com/pytorch-inference:1.7.1-cpu-py36-ubuntu18.04 |v1.1 |
107108

108109

109110
Elastic Inference Containers
@@ -129,6 +130,9 @@ Prior General Framework Container Versions
129130

130131
| Framework |Job Type |Horovod Options |CPU/GPU |Python Version Options |Example URL |
131132
|---------------------------------------------|------------------------|---------------------------------|------------|---------------------------------------|----------------------------------------------------------------------------------------------------|
133+
|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 |
134+
|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 |
135+
|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 |
132136
|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 |
133137
|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 |
134138
|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 |

0 commit comments

Comments
 (0)