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 9, Amazon Linux)

Comes with deep learning frameworks configured with CUDA 9. Includes Apache MXNet, TensorFlow, PyTorch, Keras 2.0 and Caffe2. THIS AMI IS NOT UPDATED ANYMORE. For latest AMI, visit <a href="" target="_blank"></a> Release tags/Branches used: Apache MXNet 1.1 (with Gluon) Caffe2 0.8.1... See more

Customer Reviews

Create Your Own Review

gpu is not supported in screen sessions

  • By -hj
  • on 03/15/2018

If you activate the tensorflow_p36 environment, and then run a python script that uses tensorflow, you see that the gpu is utilized. However, if you start a screen session, then activate the tensorflow_p36 environment, running the same script will only run on the CPU. Very aggravating when you need to train large scale models that take days (requiring the use of screen/tmux)

Looks promising

  • By kay
  • on 11/09/2017

Excited to start working with P3 instances with their increased GPU performance. Eagerly awaiting Keras support! Thanks for developing these AMIs.

Thx, still not yet fully operational

  • By Herve Nivon
  • on 10/26/2017

Launching went smoothly, still:

> We definitely miss Keras 2.
> Mentioned is not where you mentioned. It is located here: /home/ec2-user/
> Met some problems updating python2 packages (force to manually remove pip from /usr/lib/python2.7/dist-packages/ and reinstall it with the traditional sudo python2.7

You rock, waiting for AMI updates!

showing 1 - 3