FROM nvidia/cuda:8.0-devel

MAINTAINER Lucas Frérot <lucas.frerot@epfl.ch>

# Install any needed packages from ubuntu repos
RUN apt-get -qq update && apt-get install -y \
    build-essential \
    clang-5.0 \
    cmake \
    git \
    libboost-dev \
    libpcre3-dev \
    libpython3-dev \
    python3 \
    python3-dev \
    python3-numpy \
    python3-pytest \
    python3-scipy \
    scons \
    wget \
  && rm -rf /var/lib/apt/lists/*

# Compiling fftw
RUN cd /opt && \
 wget -O fftw.tar.gz http://www.fftw.org/fftw-3.3.7.tar.gz && \
 tar -xvzf fftw.tar.gz && \
 cd fftw-3.3.7 && \
 ./configure --prefix=$PWD --enable-shared --enable-openmp && \
 make && make install && \
 cd ../..

ENV FFTW_ROOT /opt/fftw-3.3.7

# No need for python2
RUN ln -s $(which python3) /usr/local/bin/python

# Define environment variable
ENV CUDA_ROOT /usr/local/cuda