Sign in
Your Saved List Become a Channel Partner Sell in AWS Marketplace Amazon Web Services Home Help

AWS Deep Learning Containers for PyTorch

AWS Deep Learning Containers for PyTorch

By: Amazon Web Services Latest Version: 1.6

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.



Operating System


Delivery Methods

  • Container

Pricing Information

Usage Information

Support Information

Customer Reviews