Deep Learning AMI
Developer Guide

Deep Learning AMI with Conda

The Conda DLAMI uses Anaconda virtual environments. These environments are configured to keep the different framework installations separate. It also makes it easy to switch between frameworks. This is great for learning and experimenting with all of the frameworks the DLAMI has to offer. Most users find that the new Deep Learning AMI with Conda is perfect for them.

These "Conda" AMIs will be the primary DLAMIs. It will be updated often with the latest versions from the frameworks, and have the latest GPU drivers and software. It will be generally referred to as the AWS Deep Learning AMI in most documents.

  • The Ubuntu 18.04 DLAMI has the following frameworks: Apache MXNet, Chainer, PyTorch, TensorFlow, and TensorFlow 2

  • The Ubuntu 16.04 and Amazon Linux DLAMI has the following frameworks: Apache MXNet, Caffe, Caffe2, Chainer, CNTK, Keras, PyTorch, TensorFlow, TensorFlow 2, and Theano

  • The Amazon Linux 2 DLAMI has the following frameworks: Apache MXNet, Chainer, PyTorch, TensorFlow, TensorFlow 2, and Keras

Stable versus Release Candidates

The Conda AMIs use optimized binaries of the most recent formal releases from each framework. Release candidates and experimental features are not to be expected. The optimizations depend on the framework's support for acceleration technologies like Intel's MKL DNN, which will speed up training and inference on C5, C4, and C3 CPU instance types. The binaries are also compiled to support advanced Intel instruction sets including, but not limited to AVX, AVX-2, SSE4.1, and SSE4.2. These accelerate vector and floating point operations on Intel CPU architectures. Additionally, for GPU instance types, the CUDA and cuDNN will be updated with whichever version the latest official release supports.

The Deep Learning AMI with Conda automatically installs the most optimized version of the framework for your EC2 instance upon the framework's first activation. For more information, refer to Using the Deep Learning AMI with Conda.

If you want to install from source, using custom or optimized build options, the Deep Learning Base AMI's might be a better option for you.

CUDA Support

The Deep Learning AMI with Conda's CUDA version and the frameworks supported for each:

  • CUDA 10.1 with cuDNN 7: Apache MXNet

  • CUDA 10 with cuDNN 7: PyTorch, TensorFlow, TensorFlow 2, Apache MXNet, Caffe2, Chainer

  • CUDA 9 with cuDNN 7: CNTK, Keras, Theano

  • CUDA 8 with cuDNN 6: Caffe

Specific framework version numbers can be found in the DLAMI: Release Notes

Choose this DLAMI type or learn more about the different DLAMIs with the Next Up option.

Next Up

Deep Learning Base AMI

Related Topics