AWS Machine Learning Blog

Deep Learning AMI for Ubuntu v1.3_Apr2017 Now Supports Caffe2

We are excited to announce that the AWS Deep Learning AMI for Ubuntu now supports the newly launched Caffe2 project led by Facebook. AWS is the best and most open place for developers to run deep learning, and the addition of Caffe2 adds yet another choice. To learn more about Caffe2, check out the the Caffe2 developer site or the GitHub repository.


The Deep Learning AMI v1.3_Apr2017 for Ubuntu provides a stable, secure, and high-performance execution environment for deep learning applications running on Amazon EC2. This AMI includes the following framework versions:

  • MXNet v0.9.3
  • Caffe2 v0.6.0 (new)
  • TensorFlow v1.0.1 (updated)
  • Caffe rc5
  • Theano rel-0.8.2
  • Keras 1.2.2
  • CNTK v2.0 RC1 (updated)
  • Torch master branch

The Deep Learning AMI includes the following:

  • Jupyter notebook with Python 2.7 and Python 3.4 kernels
  • Matplotlib
  • scikit-image
  • CppLint
  • Pylint
  • pandas
  • Graphviz
  • Bokeh Python packages
  • Boto and Boto 3
  • The AWS command line interface (CLI)

The Deep Learning AMI also includes the Anaconda 2 and Anaconda 3 Data Science platforms.

About the AWS Deep Learning AMIs

The AWS Deep Learning AMI (available for Amazon Linux and Ubuntu) and the AWS Deep Learning CloudFormation Template let you quickly deploy and run any of the major deep learning frameworks at any scale. The AWS Deep Learning AMI lets you create managed, auto-scaling clusters of GPUs for large scale training, and run inference on trained models. It is pre-installed with Apache MXNet, TensorFlow, Caffe, Theano, Torch, CNTK, and Keras. The AWS Deep Learning AMI is provided and supported by Amazon Web Services, for use on Amazon EC2. There is no additional charge for the AWS Deep Learning AMI – you only pay for the AWS resources needed to store and run your applications.

If you have questions or suggestions, please leave a comment below.