Dockerfile

# Dockerfile

# docker build .

# Based on CUDA11.0 & CuDNN8
FROM nvidia/cuda:10.2-devel-ubuntu18.04

# Install Non-GPU Dependencies.
RUN apt update --allow-unauthenticated && version="7.0.0-1+cuda10.2" ; \
    apt install -y \
    libnvinfer7=${version} libnvonnxparsers7=${version} libnvparsers7=${version} \
    libnvinfer-plugin7=${version} libnvinfer-dev=${version} libnvonnxparsers-dev=${version} \
    libnvparsers-dev=${version} libnvinfer-plugin-dev=${version} python-libnvinfer=${version} \
    python3-libnvinfer=${version} && \
    apt-mark hold \
    libnvinfer7 libnvonnxparsers7 libnvparsers7 libnvinfer-plugin7 libnvinfer-dev libnvonnxparsers-dev libnvparsers-dev libnvinfer-plugin-dev python-libnvinfer python3-libnvinfer

# Install OpenCV Dependencies
RUN apt install -y software-properties-common || apt install -y software-properties-common && \
    add-apt-repository "deb http://security.ubuntu.com/ubuntu xenial-security main" && \
    APT_DEPS="git cmake libgtk-3-dev libavcodec-dev libavformat-dev libswscale-dev libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev x264 v4l-utils python3-dev python3-pip libcanberra-gtk-module libcanberra-gtk3-module" && \
    apt install -y $APT_DEPS || apt install -y $APT_DEPS && \
    python3 -m pip install numpy

# Compile OpenCV
RUN git clone --branch 4.4.0 https://github.com/opencv/opencv.git && \
    cd opencv && mkdir build && cd build && \
    cmake .. -DCMAKE_BUILD_TYPE=Release \
             -DCMAKE_INSTALL_PREFIX=/usr/local \
             -DWITH_TBB=ON \
             -DWITH_V4L=ON \
             -DBUILD_TESTS=OFF \
             -DBUILD_OPENCV_PYTHON3=OFF && \
    make -j && make install

# Install HyperPose Dependencies
RUN apt install -y python3-dev python3-pip subversion libgflags-dev

COPY . /hyperpose

# Download related data
RUN for file in $(find /hyperpose/scripts -type f -iname 'download*.sh'); do sh $file; done

# Build Repo
RUN cd hyperpose && mkdir build && cd build && \
    cmake .. -DCMAKE_BUILD_TYPE=Release && make -j

WORKDIR /hyperpose/build

ENTRYPOINT ["./hyperpose-cli"]