#FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
MAINTAINER Suren A. Chilingaryan <csa@suren.me>

# Install Devel & Python
RUN apt-get update -qq \
 && apt-get install --no-install-recommends -y \
    git pkg-config netcat supervisor \
    telnet iputils-ping dnsutils \
    python python-dev python-pip python-setuptools python-virtualenv python-wheel python-numpy python-scipy python-h5py python-yaml python-pydot python-matplotlib libopenblas-base

# Install Keras + Tensor Flow
ARG TENSORFLOW_VERSION=1.12.0
ARG TENSORFLOW_DEVICE=gpu
ARG TENSORFLOW_APPEND=_gpu

ARG KERAS_VERSION=2.2.4
ENV KERAS_BACKEND=tensorflow

RUN pip --no-cache-dir install \
    https://storage.googleapis.com/tensorflow/linux/${TENSORFLOW_DEVICE}/tensorflow${TENSORFLOW_APPEND}-${TENSORFLOW_VERSION}-cp27-none-linux_x86_64.whl

RUN pip --no-cache-dir install --no-dependencies \
    git+https://github.com/fchollet/keras.git@${KERAS_VERSION}

ADD tests/ /root/tests

# install dependencies from debian packages
RUN apt-get install --no-install-recommends -y \
    mc \
    libsm6 libxrender-dev \
    openmpi-bin openmpi-doc libopenmpi-dev \
    python-gdcm python-mysqldb

# upgrade pip
RUN pip --no-cache-dir install --upgrade pip

# install dependencies from python packages
RUN pip --no-cache-dir install \
    setuptools==39.1.0 numpy==1.14.5 tifffile==0.14.0 SimpleParse==2.1.1 redis==2.10.6 rq==0.12 \
    scipy weave scikit-image opencv-python h5py pydicom Pillow django mpi4py pycuda

COPY supervisor/*.conf /etc/
COPY *-entrypoint.sh /
COPY run-*.sh /
ENTRYPOINT ["/docker-entrypoint.sh"]

ENV BIOMEDISA_DEVICE=${TENSORFLOW_DEVICE}
CMD /usr/bin/supervisord -c /etc/supervisord-${BIOMEDISA_DEVICE}.conf