Sign in
Migration Mapping Assistant Your Saved List Partners Sell in AWS Marketplace Amazon Web Services Home Help

Deep Learning AMI with Source Code (CUDA 8, Ubuntu)

Comes with deep learning frameworks configured with CUDA 8. Includes Apache MXNet, Caffe, Caffe2, TensorFlow, PyTorch, Theano, CNTK, Torch and Keras. THIS AMI IS NOT UPDATED ANYMORE. For latest AMI, go to: <a href="" target="_blank"></a> Release tags/Branches used: MXNet 0.11.0 TensorFlow... See more

Customer Reviews

Create Your Own Review

Keras, tensorflow etc. not installed

  • By PD
  • on 01/05/2018

Like others have noted, Keras, Tensorflow are missing and Amazon support wants me to pay to get it sorted out. No thanks!

Great AMI with Python3

  • By yoshitaka
  • on 10/17/2017

As others pointed out, use python3.
Most of the major frameworks are installed with GPU enabled.
Of course we can instantly use the Gluon.

Great AMI

  • By -m
  • on 09/28/2017

Saves lots of time in searching for CUDA drivers, cuDNN, and building DL frameworks. New Keras support for MxNet is great too!

Deep Learning AMI is just an Ubuntu image with nothing installed :)

  • By Janosch
  • on 09/27/2017

Two weeks ago I got excited at a presentation from two AWS architects in Cambridge telling us how great it is to use Amazon cloud services because they come with everything pre-installed and you could focus on doing your research.

I did choose the p2.xlarge AMI in Ireland to do deep learning on image data. I tried following several tutorials I found on Amazon websites and smart IT blogs to be successful. The disappointment started with Jupyter not being installed. Then trying to do simple Python import commands resulted in Tensorflow not found, MXnet not found and including other core packages required.
After installing Jupyter and lots of other libraries needed to get tensorflow-gpu support to work it turned out that the Cuda library is not properly installed. See also other reviews.

I have spent now a full day installing stuff from scratch which cost me a lot of time and 10$ in cloud services without much progress. I am not going to give up and will soon be a deep learning professional IT architect. Thank you Amazon for your enticing marketing and despicable software services! Lesson learned :)

Getting started REALLY required

  • By Frustrated
  • on 09/18/2017

I need to get the following:
- CUDA (8.0)
- cuDNN (6)
- Tensorflow (latest)

It is extremely unclear how to get started activating just that.
Also - I may be mistaken, but only tensorflow was installed, not tensorflow-gpu, so I really don't see the point in the pre-installed environments here...

Deep Learning AMI Ubuntu: Free tier or not Free tier

  • By Vlad
  • on 09/13/2017

Tried to launch the instance in subj, which is marked as 'free tier eligible' and got the message - '...not eligible for free usage tier'. Bug? User error? else?

Got Tensorflow working via Tensorflow3

  • By ML VP
  • on 08/28/2017

Per the other reviews, the default tensorflow in ~/src/tensorflow was problematic.

However, we've had no trouble using the Python3+ compilation, even with Python2.7:

It's a reasonable work-around until an updated AMI is released.

Tensorflow does not work with Jupyter

  • By Jordan
  • on 08/10/2017

import tensorflow fails from a jupyter notebook in both python2 and python3. It was successful on the python command line, however.

Cannot load tensorflow

  • By Sean
  • on 08/09/2017

Got it up and running and ran "import tensorflow" -- no luck though.

Traceback (most recent call last):
File "/home/ubuntu/deployments/data-pipelines/scratch/", line 1, in
import tensorflow as tf
File "/usr/local/lib/python2.7/dist-packages/tensorflow/", line 24, in
from tensorflow.python import *
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/", line 49, in
from tensorflow.python import pywrap_tensorflow
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/", line 52, in
raise ImportError(msg)
ImportError: Traceback (most recent call last):
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/", line 41, in
from tensorflow.python.pywrap_tensorflow_internal import *
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/", line 28, in
_pywrap_tensorflow_internal = swig_import_helper()
File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/", line 24, in swig_import_helper
_mod = imp.load_module('_pywrap_tensorflow_internal', fp, pathname, description)
ImportError: cannot open shared object file: No such file or directory

Failed to load the native TensorFlow runtime.


for some common reasons and solutions. Include the entire stack trace
above this error message when asking for help.

Process finished with exit code 1

Tensorflow doesn't work.

  • By DLResearcher
  • on 07/27/2017

I am using this AMI with a p2 xlarge instance. When I try to test tensorflow, I got the following message:

ImportError: cannot open shared object file: No such file or directory

Failed to load the native TensorFlow runtime.