FROM nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04

RUN apt update -y && apt install -y python python-pip git cmake
RUN pip install theano numpy tqdm cython markupsafe
RUN git clone https://github.com/Theano/libgpuarray.git && \
    cd libgpuarray && \
    mkdir Build && \
    cd Build && \
    cmake .. -DCMAKE_BUILD_TYPE=Release && \
    make && \
    make install && \
    cd .. && \
    python setup.py build && \
    python setup.py install && \
    ldconfig

