Dockerfile

# Dockerfile created from https://github.com/keras-team/keras/tree/master/docker.

ARG cuda_version=9.0
ARG cudnn_version=7
FROM nvidia/cuda:${cuda_version}-cudnn${cudnn_version}-devel

# Install system packages
RUN apt-get update && apt-get install -y --no-install-recommends \
      bzip2 \
      g++ \
      git \
      graphviz \
      libgl1-mesa-glx \
      libhdf5-dev \
      openmpi-bin \
      wget \
      vim && \
    rm -rf /var/lib/apt/lists/*

# Install conda
ENV CONDA_DIR /opt/conda
ENV PATH $CONDA_DIR/bin:$PATH

RUN wget --quiet --no-check-certificate https://repo.continuum.io/miniconda/Miniconda3-4.2.12-Linux-x86_64.sh && \
    echo "c59b3dd3cad550ac7596e0d599b91e75d88826db132e4146030ef471bb434e9a *Miniconda3-4.2.12-Linux-x86_64.sh" | sha256sum -c - && \
    /bin/bash /Miniconda3-4.2.12-Linux-x86_64.sh -f -b -p $CONDA_DIR && \
    rm Miniconda3-4.2.12-Linux-x86_64.sh && \
    echo export PATH=$CONDA_DIR/bin:'$PATH' > /etc/profile.d/conda.sh

# Install Python packages and keras
#ENV NB_USER keras
#ENV NB_UID 1000

#RUN useradd -m -s /bin/bash -N -u $NB_UID $NB_USER && \
#    chown $NB_USER $CONDA_DIR -R && \
#    mkdir -p /src && \
#    chown $NB_USER /src

#USER $NB_USER

ARG python_version=3.6

RUN conda install -y python=${python_version} && \
    pip install --upgrade pip && \
    pip install \
      sklearn_pandas \
      tensorflow-gpu \
      Cython \
      scikit-image \
      opencv-python \
      six \
      imgaug \
      librosa && \
    conda install \
      bcolz \
      h5py \
      matplotlib \
      mkl \
      nose \
      notebook \
      Pillow \
      pandas \
      pydot \
      pygpu \
      pyyaml \
      scikit-learn && \
#     theano && \
    git clone git://github.com/keras-team/keras.git /keras && pip install -e /keras[tests] && \
    pip install git+git://github.com/keras-team/keras.git && \
    conda clean -yt

#ADD theanorc /home/keras/.theanorc

RUN pip install pycocotools

ENV PYTHONPATH='/keras/:$PYTHONPATH'

RUN mkdir /workspace

WORKDIR /workspace

EXPOSE 8888

CMD jupyter notebook --port=8888 --ip=0.0.0.0 --allow-root