Features of DLAMI
The features of Amazon 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:
-
Deep Learning AMI with Conda – Frameworks installed separately using
conda
packages and separate Python environments. -
Deep Learning Base AMI – No frameworks installed; only NVIDIA CUDA
and other dependencies.
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
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.