Amazon just recently released new AWS EC2 P2 instance types featuring K80 GPUs. In this blog post I’ll cover how to configure and install TensorFlow on a Ubuntu 16.04 targeting these new GPU instance types. This article is building upon previous blog post by John Ramey on installing TensorFlow for older AWS GPU instance types.
The following components will be installed:
- CUDA 8.0
- Download cuDNN v5.1 (August 10, 2016), for CUDA 8.0
- Bazel 0.3.2 (Java 8 is a dependency)
- TensorFlow 0.11
This tutorial was written for a p2.xlarge instance type, however all AWS P2 instance types should work. You’re going to need 16 GiB of storage on the root partition to follow this guide.