Dockerfile for Cuda/Ubuntu based container that runs VSR algorithm iSeeBetter

iSeeBetter Nvidia Cuda Dockerfile

FROM nvidia/cuda:11.1-devel-ubuntu18.04

ARG DEBIAN_FRONTEND=noninteractive

LABEL maintainer="Isaac Spiegel"

WORKDIR /

RUN set -ex \
    && apt-get update \
    && apt-get install -y software-properties-common \
    && apt-get install -y git \
    && apt-get install -y curl

RUN set -ex \
    && add-apt-repository universe \
    && add-apt-repository multiverse \
    && apt-get update \ 
    && apt-get install -y --no-install-recommends \
      beignet-opencl-icd \
      mesa-opencl-icd \
      ocl-icd-libopencl1 \
      libopencv-core3.2 \
      libopencv-imgcodecs3.2 \
      ocl-icd-opencl-dev \
    && rm -rf /var/lib/apt/lists/*

RUN set -ex \
    && mkdir -p /usr/src \
    && cd /usr/src \
    && git clone https://github.com/amanchadha/iSeeBetter 

# Install python3
RUN set -ex \
    && apt install -y python3 \
    && apt update \
    && apt install -y python3-distutils \
    && curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
    && python3 get-pip.py \
    && rm get-pip.py 

# Some (python package) dependency versions no longer offered from requirements.txt so manual install
RUN set -ex \
    && pip3 install \
        torch \
        torchvision \
        pytorch-ssim==0.1 \
        scikit-image==0.15.0 \
        tqdm==4.37.0 \
        opencv-python \
        Cython \
        numpy

RUN set -ex \
    && apt install -y --no-install-recommends build-essential gcc g++ python3-dev

RUN set -ex \
    && cd /usr/src/iSeeBetter/pyflow \
    && pip3 install Cython numpy \
    && python3 setup.py build_ext -i \
    && cp ./pyflow*.so ..

ADD vid4-dataset /usr/src/iSeeBetter/Vid4

WORKDIR /usr/src/iSeeBetter

ENTRYPOINT ["python3","iSeeBetterTest.py"]