The AWS Deep Learning AMIs for Ubuntu and Amazon Linux now come with newer versions of the following deep learning frameworks: PyTorch 1.0 , MXNet 1.3.1 , and Chainer 5.1 . This release also includes upgrades of the NVIDIA stack, including CUDA 10, cuDNN 7.4.1, and NCCL 2.3.7. For GPU-based training, PyTorch 1.0 in the AMIs is fully configured to take advantage of performance improvements in CUDA 10.