CUDA and Tensorflow in Docker

Mon, 24 Jul 2017 12:00:25 +0000

In this howto we will get CUDA working in Docker. And - as bonus - add Tensorflow on top! However, please note that you’ll need following prereqs:

GNU/Linux x86_64 with kernel version > 3.10
Docker >= 1.9 (official docker-engine, docker-ce or docker-ee only)
NVIDIA GPU with Architecture > Fermi (2.1)
NVIDIA drivers >= 340.29 with binary nvidia-modprobe

We will install the NVIDIA drivers in this tutorial, so you should only have the right kernel and docker version already installed, we’re using a Ubuntu 15.05 x64 machine here. For CUDA, you’ll need a Fermi 2.1 CUDA card (or better), for tensorflow a >= 3.0 CUDA card…

Tensorflow on Nico Maas

CUDA and Tensorflow in Docker