AWS Deep Learning Containers

AWS Deep Learning Containers

One-stop containers for AI/ML on AWS

Description

Running deep learning in the cloud can eat a day just matching CUDA, framework and driver versions. AWS Deep Learning Containers are Docker images with deep learning frameworks preinstalled and optimized for AWS.

They cover PyTorch, TensorFlow and major inference engines, ready for SageMaker, EKS and EC2.

Features



Frameworks:PyTorch and TensorFlow.

Compatibility:CUDA set up.

Optimized:Tuned for AWS.

Services:SageMaker, EKS and EC2.