FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
MAINTAINER Vladislav Belavin <belavin@phystech.edu>

# basic setup
SHELL ["/bin/bash", "-c"]

# update
RUN apt-get -y update && \
    apt-get install -y --no-install-recommends software-properties-common && \
    apt-get -y update && \
    apt-get install -y --no-install-recommends --allow-change-held-packages apt-utils curl \
    bzip2 gcc git wget g++ build-essential zip unzip libc6-dev make pkg-config  \
    libcudnn7 libnccl2 libnccl-dev && \
    apt-get clean -y && rm -rf /var/lib/apt/lists/*

# install miniconda + scientific python stack
RUN curl -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
    bash Miniconda3-latest-Linux-x86_64.sh -b -p /root/miniconda && rm Miniconda3-latest-Linux-x86_64.sh
# activate
ENV PATH /root/miniconda/bin:$PATH

RUN conda update -n base conda && pip install --upgrade pip && conda install python=3.6 && \
    conda config --add channels conda-forge && \
    conda install -y pip numpy scikit-learn scipy jupyter matplotlib tqdm pandas tqdm psutil Cython numpy pyyaml setuptools cmake cffi typing && \
    conda install pytorch torchvision cudatoolkit=9.0 -c pytorch && \
    conda clean --all -y && \
    rm -rf ~/.cache/pip

RUN pip install pyro-ppl schema timeout_decorator

WORKDIR /root

RUN jupyter notebook --generate-config

USER root
EXPOSE 8888

WORKDIR /root/workdir
ENTRYPOINT ["jupyter", "notebook", "--ip=0.0.0.0", "--port=8888", "--allow-root"]