Dockerfile

FROM tensorflow/tensorflow:1.4.1-gpu-py3
# According to Dockerfile for above:
# https://github.com/tensorflow/tensorflow/blob/v1.4.1/tensorflow/tools/docker/Dockerfile.gpu
#   FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04
#   ...

LABEL description="Prebuilt jupyter environment based on official GPU tensorflow docker image."
LABEL maintainer="https://github.com/rlan/notebooks"

# Build-time metadata as defined at http://label-schema.org
ARG BUILD_DATE
ARG VCS_REF
ARG VERSION
LABEL org.label-schema.build-date=$BUILD_DATE \
      org.label-schema.vcs-ref=$VCS_REF \
      org.label-schema.vcs-url="https://github.com/rlan/notebooks" \
      org.label-schema.version=$VERSION \
      org.label-schema.schema-version="1.0"

RUN apt-get -qq update \
    && apt-get -qq install -y --no-install-recommends \
        graphviz \
        libhdf5-dev \
        vim \
        wget \
    && apt-get -qq clean \
    && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*

RUN pip --no-cache-dir install -q --upgrade pip \
    && pip --no-cache-dir install -q -U \
        h5py \
        jupyterlab \
        keras \
        opencv-python-headless \
        pydot \
        scikit-image

# TensorBoard
EXPOSE 6006
# IPython
EXPOSE 8888

WORKDIR /notebooks