

AWS Deep Learning Containers for PyTorch
Linux/Unix
Linux/Unix
Product Overview
The AWS Deep Learning Containers for PyTorch include containers for training and inference for CPU and GPU, optimized for performance and scale on AWS. These Docker images have been tested with SageMaker, EC2, ECS, and EKS and provide stable versions of NVIDIA CUDA, cuDNN, Intel MKL, Horovod and other required software components to provide a seamless user experience for deep learning workloads. All software components in these images are scanned for security vulnerabilities and updated or patched in accordance with AWS Security best practices.
Version
1.6
Operating System
Linux
Delivery Methods