Dockerfile.keras

FROM nvidia/cuda:8.0-cudnn5-devel-ubuntu16.04
MAINTAINER Luke Yeager <lyeager@nvidia.com>

RUN apt-get update \
    && apt-get install -y --no-install-recommends \
        build-essential \
        ca-certificates \
        curl \
        git \
        emacs \
        libhdf5-dev \
        libopenblas-dev \
        python-all-dev \
        python-h5py \
        python-numpy \
        python-pip \
        python-scipy \
        python-setuptools \
        python-wheel \
        python-yaml \
        sudo \
        vim \
        wget \
    && rm -rf /var/lib/apt/lists/*

RUN pip install theano tensorflow keras