FROM nvidia/cuda:8.0-devel-ubuntu14.04

ENV PATH="/usr/local/cuda/bin:"$PATH
ENV LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda/lib:/usr/local/lib:"$LD_LIBRARY_PATH

RUN apt-get update --fix-missing && \
    apt-get install -y python-pip python-dev git software-properties-common && \
    apt-get clean autoclean && \
    apt-get autoremove -y
RUN pip install --upgrade pip
RUN pip install virtualenv Pillow

# this sets up a python3 virtualenv and activates it
ARG python_version=2
RUN if [ $python_version -eq 3 ]; then \
    apt-get install -y python3 python3-pip && \
    apt-get clean autoclean && \
    apt-get autoremove -y && \
    pip3 install virtualenv && \
    virtualenv -p python3 /tmp/env3; fi
ENV PATH="/tmp/env3/bin:$PATH"

RUN apt-get install -y bzip2 wget coreutils libjasper1 libjpeg8 libpng12-0 curl libsox-fmt-all libsox-dev sox libopencv-dev python-opencv unzip clang libcurl4-openssl-dev cmake numactl

RUN apt-get install -y bsdmainutils

# Install hw detect packages
RUN apt-get update --fix-missing && apt-get -y install dmidecode numactl pciutils linux-cloud-tools-generic linux-tools-generic net-tools lm-sensors cpuid

# Get and build Open MPI
RUN wget -q https://www.open-mpi.org/software/ompi/v1.10/downloads/openmpi-1.10.3.tar.gz && \
    tar -xzvf ./openmpi-1.10.3.tar.gz && \
    cd openmpi-1.10.3 && \
    ./configure --prefix=/usr/local/mpi && \
    make -j all && \
    sudo make install && cd .. && \
    rm -rf openmpi-1.10.3 openmpi-1.10.3.tar.gz

# Add Open MPI to path
ENV PATH="/usr/local/mpi/bin:$PATH"
ENV LD_LIBRARY_PATH="/usr/local/mpi/lib:$LD_LIBRARY_PATH"
ENV MPI_ROOT="/usr/local/mpi"