Dockerfile.py2-th-gpu

# docker-keras - Keras in Docker with Python 2 and Theano on GPU

FROM gw000/debian-cuda:9.1_7.0
MAINTAINER gw0 [http://gw.tnode.com/] <gw.2018@ena.one>

# install debian packages
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update -qq \
 && apt-get install --no-install-recommends -y \
    # install essentials
    build-essential \
    g++ \
    git \
    openssh-client \
    # install python 2
    python \
    python-dev \
    python-pip \
    python-setuptools \
    python-virtualenv \
    python-wheel \
    pkg-config \
    # requirements for numpy
    libopenblas-base \
    python-numpy \
    python-scipy \
    # requirements for keras
    python-h5py \
    python-yaml \
    python-pydot \
    # requirements for theano
    cmake \
    cython \
 && apt-get clean \
 && rm -rf /var/lib/apt/lists/*

# build and install requirements for theano
RUN git clone https://github.com/Theano/libgpuarray.git /tmp/libgpuarray \
 && cd /tmp/libgpuarray \
 && mkdir ./Build \
 && cd ./Build \
 && cmake .. -DCMAKE_BUILD_TYPE=Release \
 && make \
 && make install \
 && cd .. \
 && python setup.py build \
 && python setup.py install \
 && cd / \
 && rm -rf /tmp/libgpuarray \
 && ldconfig

ARG THEANO_VERSION=1.0.1
ENV THEANO_FLAGS='device=cuda,floatX=float32'
RUN pip --no-cache-dir install git+https://github.com/Theano/Theano.git@rel-${THEANO_VERSION}

ARG KERAS_VERSION=2.1.4
ENV KERAS_BACKEND=theano
RUN pip --no-cache-dir install --no-dependencies git+https://github.com/fchollet/keras.git@${KERAS_VERSION}

# quick test and dump package lists
RUN python -c "import theano; print(theano.__version__)" \
 && dpkg-query -l > /dpkg-query-l.txt \
 && pip2 freeze > /pip2-freeze.txt

WORKDIR /srv/