Select your cookie preferences

We use essential cookies and similar tools that are necessary to provide our site and services. We use performance cookies to collect anonymous statistics, so we can understand how customers use our site and make improvements. Essential cookies cannot be deactivated, but you can choose “Customize” or “Decline” to decline performance cookies.

If you agree, AWS and approved third parties will also use cookies to provide useful site features, remember your preferences, and display relevant content, including relevant advertising. To accept or decline all non-essential cookies, choose “Accept” or “Decline.” To make more detailed choices, choose “Customize.”

Features of DLAMI

Focus mode
Features of DLAMI - AWS Deep Learning AMIs

The features of AWS Deep Learning AMIs (DLAMI) include preinstalled deep learning frameworks, GPU software, model servers, and model visualization tools.

Preinstalled frameworks

There are currently two primary flavors of DLAMI with other variations related to the operating system (OS) and software versions:

The Deep Learning AMI with Conda uses conda environments to isolate each framework, so you can switch between them at will and not worry about their dependencies conflicting. The Deep Learning AMI with Conda supports the following frameworks:

  • PyTorch

  • TensorFlow 2

Note

DLAMI no longer supports the following deep learning frameworks: Apache MXNet, Microsoft Cognitive Toolkit (CNTK), Caffe, Caffe2, Theano, Chainer, and Keras.

Preinstalled GPU software

Even if you use a CPU-only instance, the DLAMIs will have NVIDIA CUDA and NVIDIA cuDNN. The installed software is the same regardless of the instance type. Keep in mind that GPU-specific tools work only on an instance that has at least one GPU. For more information about instance types, see Choosing a DLAMI instance type.

For more information about CUDA, see CUDA Installations and Framework Bindings.

Model serving and visualization

Deep Learning AMI with Conda comes preinstalled with model servers for TensorFlow, as well as TensorBoard for model visualizations. For more information, see TensorFlow Serving.

PrivacySite termsCookie preferences
© 2025, Amazon Web Services, Inc. or its affiliates. All rights reserved.