AWS Deep Learning Containers for PyTorch Training v2.1
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