ARG REPOSITORY=nvidia/cuda
FROM $REPOSITORY:8.0-runtime-ubuntu16.04
LABEL maintainer "NVIDIA CORPORATION <cudatools@nvidia.com>"

RUN echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list

ENV CUDNN_VERSION 6.0.21
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}"

RUN apt-get update && apt-get install -y --no-install-recommends \
            libcudnn6=$CUDNN_VERSION-1+cuda8.0 && \
    rm -rf /var/lib/apt/lists/*

RUN apt-get update
RUN apt-get install -y python3-pip python3-dev
RUN apt-get install -y python3-tk
RUN pip3 install --upgrade pip
RUN pip3 install keras
RUN pip3 install tensorflow-gpu==1.3.0
RUN pip3 install http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp35-cp35m-manylinux1_x86_64.whl
RUN pip3 install torchvision
RUN pip3 install scikit-learn
RUN pip3 install sk-video
RUN pip3 install scikit-image
RUN pip3 install pandas
RUN pip3 install tqdm
RUN pip3 install SimpleITK
RUN pip3 install pydicom
RUN pip3 install h5py