nvdocker-keras

Requirements

• Docker CE for Ubuntu
• nvidia-docker
• CUDA

Running the container

Using Makefile to simplify docker commands within make commands.

build image

$ make build

start container

$ make start

attach container

$ make attach

stop container

$ make stop

clean container and images

$ make clean

Keras example

$ make build
$ make start
$ make attach
$ git clone https://github.com/fchollet/keras.git
$ cd keras/examples
$ python mnist_cnn.py