AWS Machine Learning Blog

The AWS Deep Learning AMI for Ubuntu is Now Available with CUDA 8, Ubuntu 16, and the Latest Versions of Deep Learning Frameworks

The AWS Deep Learning AMI lets you build and scale deep learning applications in the cloud, at any scale. The AMI comes pre-installed with popular deep learning frameworks, to let you to train sophisticated, custom AI models, experiment with new algorithms, or to learn new skills and techniques. The latest release of the AWS Deep Learning AMI Ubuntu Version includes several notable updates that help accelerate development of high-performance algorithms. The AMI now comes bundled with the Ubuntu 16.04 base image, NVIDIA CUDA 8 drivers, and the following deep learning framework versions:

  • MXNet v0.10.0.post1
  • TensorFlow v1.2.0
  • Theano 0.9.0
  • Caffe v1.0
  • Caffe2 v0.7.0
  • Keras 1.2.2
  • CNTK v2.0
  • Torch (master branch)

The AMI Ubuntu Version is now live in seven AWS Regions:

  • us-east-2 (Ohio)
  • us-east-1 (N. Virginia)
  • us-west-2 (Oregon)
  • us-west-1 (N. California)
  • ap-southeast-2 (Sydney)
  • ap-northeast-1 (Tokyo)
  • ap-northeast-2 (Seoul)

Both the Ubuntu and Amazon Linux versions of the AMI are now easier to locate in the EC2 console. As you configure your instance, you can choose an AMI in the Quick Start in Step 1, which lists commonly used instances.

You can find the instances as you scroll though the Quick Start list—it’s the sixth from the top:

For more information, see the instructions for accessing your instance and launching a Jupyter notebook.

Get started with the AWS Deep Learning AMI Ubuntu Version by visiting the AWS Marketplace.  For package details, see the Apache MXNet on AWS Developer Guide.