Skip to content

AWS Deep Learning Containers for PyTorch Training v2.1

Choose a tag to compare

@arjkesh arjkesh released this 22 Jan 18:50

This tag refers to the release of AWS Deep Learning Containers for PyTorch Training version 2.1. The images below were built using the PyTorch 1.3.1 Dockerfiles at this commit of master. Subsequent Dockerfile commits to master will be incorporated in future releases.

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