FROM tensorflow/tensorflow:1.14.0-gpu-py3


RUN mkdir /physionet
COPY ./ /physionet
WORKDIR /physionet

ARG CUDA_VERSION=10-0
ARG CUDNN_VERSION=7.4.2.24

RUN apt-get update && apt-get install -y --no-install-recommends \
        build-essential \
        curl \
        git \
        libcudnn7=${CUDNN_VERSION}-1+cuda10.0 \
        libcudnn7-dev=${CUDNN_VERSION}-1+cuda10.0 \
	libnccl2=2.4.2-1+cuda10.0 \
	libnccl-dev=2.4.2-1+cuda10.0 \
        libcurl3-dev \
        libfreetype6-dev \
        libhdf5-serial-dev \
        libzmq3-dev \
        pkg-config \
        rsync \
        software-properties-common \
        unzip \
        zip \
        zlib1g-dev \
        wget \
        && \
    rm -rf /var/lib/apt/lists/* && \
    find /usr/local/cuda-10.0/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
    rm /usr/lib/x86_64-linux-gnu/libcudnn_static_v7.a

# install python 3.6 and pip


RUN apt-get update && \
        apt-get install -y software-properties-common && \
        add-apt-repository ppa:deadsnakes/ppa && \
        apt-get update -y  && \
        apt-get install -y python3-pip && \
        apt-get install -y git


RUN apt-get update && \
        python3 -m pip install pip --upgrade && \
        python3 -m pip install wheel && \
        python3 -m pip install pandas==0.24.2 && \
        python3 -m pip install scipy==1.2.1 && \
        python3 -m pip install numpy==1.16.4 && \
        python3 -m pip install keras==2.2.4 && \
		python3 -m pip install keras-multi-head && \
		python3 -m pip install scikit-learn==0.21.3

CMD ["/bin/bash"]


