Dockerfile

FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04
WORKDIR /fantastic-filter-cli
ENV TF_CPP_MIN_LOG_LEVEL 3
ENV FF_MODEL_PATH /fantastic-filter-cli/FF_Model
ENV CUDA_CACHE_MAXSIZE 2147483648

RUN apt-get update && apt-get install -y gcc wget git make --no-install-recommends && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

RUN rm ./* -rf && \
    git clone https://github.com/ray1422/Fantastic-Filter-CLI.git ./ && \
    git checkout master

RUN wget https://github.com/ray1422/Fantastic-Filter-CLI/releases/download/v0.0.1-alpha/libtensorflow_cu10-1.tar.gz -O tf.tar.gz -q && \
    tar -C /usr/local -xzf tf.tar.gz && \
    ldconfig
# RUN wget https://storage.googleapis.com/tensorflow-nightly/github/tensorflow/lib_package/libtensorflow-gpu-linux-x86_64.tar.gz -O tf.tar.gz -q && \
#     tar -C /usr/local -xzf tf.tar.gz && \
#     ldconfig
# RUN wget https://storage.googleapis.com/tensorflow/libtensorflow/libtensorflow-gpu-linux-x86_64-2.3.0.tar.gz -O tf.tar.gz -q && \
#     tar -C /usr/local -xzf tf.tar.gz && \
#     ldconfig

RUN wget -O FF_Model.tar.gz https://github.com/ray1422/Fantastic-Filter-Professional-Plus/releases/download/v0.0.1/FF_Model.tar.gz -q && \
    tar -xvf FF_Model.tar.gz

RUN make main && \
    make test && \
    ln -s ${PWD}/bin/fantastic-filter-cli /bin/fantastic-filter-cli && \
    ln -s /bin/fantastic-filter-cli /bin/ff-cli