Dockerfile.cudaopencv4contrib.amd64

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

RUN sed -i -e 's/http:\/\/security/mirror:\/\/mirrors/' -e 's/\/ubuntu\//\/mirrors.txt/' /etc/apt/sources.list && \
	sed -i -e 's/http:\/\/archive/mirror:\/\/mirrors/' -e 's/\/ubuntu\//\/mirrors.txt/' /etc/apt/sources.list && \
	apt-get update && \
        apt-get install -y \
        build-essential \
        ccache \
        cmake \
        gfortran \
        git \
        libatlas-base-dev \
        libavcodec-dev \
        libavformat-dev \
        libeigen3-dev\
        libgtk2.0-dev \
        libhdf5-serial-dev \
        libjpeg-dev \
        libpng-dev \
        libpq-dev \
        libswscale-dev \
        libtbb-dev \
        libtiff-dev \
        libv4l-dev \
        pkg-config \
        unzip \
        wget \
        yasm \
        libprotobuf-dev \
        ffmpeg \
        bison \
        flex



WORKDIR /opt/
RUN git clone --depth=1 https://github.com/opencv/opencv_contrib.git && \
      git clone --depth=1 https://github.com/opencv/opencv.git && \
      mkdir opencv/build && \
      cd opencv/build && \
      cmake -DWITH_CUDA=ON -DWITH_TBB=ON -DENABLE_FAST_MATH=1 -DCUDA_FAST_MATH=1 -DWITH_CUBLAS=1 -DOPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules .. && \
      make -j$((`nproc`+5)) && \
      make install && \
      cd .. && \
      rm -rf opencv*

# OpenCV4 and CUDA installed