diff --git a/Dockerfile b/Dockerfile index f9727f2..b5e6d5b 100644 --- a/Dockerfile +++ b/Dockerfile @@ -7,181 +7,25 @@ # Last Modified: May 24, 2020 # Editor: K4YT3X -# Last Modified: May 30, 2020 +# Last Modified: June 5, 2020 +# using Ubuntu LTS 19.10 +# Ubuntu 20.x is incompatible with Nvidia libraries FROM ubuntu:19.10 -#FROM nvidia/cuda:10.1-cudnn7-devel-ubuntu18.04 +# file mainainter labels LABEL maintainer="Danielle Douglas " LABEL maintainer="Lhanjian " LABEL maintainer="K4YT3X " -# Don't ask questions during image setup. -ENV DEBIAN_FRONTEND noninteractive -ENV NASM_VERSION 2.14 -ENV NVCODEC_VERSION 8.2.15.6 -ENV PKG_CONFIG_PATH /usr/local/lib/pkgconfig -ENV FFMPEG_VERSION 4.1.2 +# run installation +RUN apt-get update \ + && apt-get install -y git-core \ + && git clone --recurse-submodules --progress https://github.com/k4yt3x/video2x.git /tmp/video2x/video2x \ + && bash -e /tmp/video2x/video2x/src/video2x_setup_ubuntu.sh / -# Install apt-fast, because we got gigs to download. -RUN apt-get update && apt-get install -y apt-utils &&\ - apt-get install -y --no-install-recommends software-properties-common &&\ - add-apt-repository ppa:apt-fast/stable &&\ - add-apt-repository -y ppa:graphics-drivers/ppa &&\ - apt-get install -y --no-install-recommends apt-fast && apt-fast update - -# Install dependencies -RUN apt-fast install -y --no-install-recommends autoconf ca-certificates curl ffmpeg frei0r-plugins-dev g++ gcc git-core gnupg2 gnutls-dev ladspa-sdk libass-dev libass-dev libbluray-dev libc6-dev libcaca-dev libcdio-paranoia-dev libchromaprint-dev libcodec2-dev libfdk-aac-dev libfdk-aac-dev libfontconfig1-dev libfreetype6-dev libfreetype6-dev libfreetype6-dev libfribidi-dev libgme-dev libgnutls28-dev libgsm1-dev libjack-dev libjack-dev liblilv-dev libmagic-dev libmagic1 libmodplug-dev libmp3lame-dev libmp3lame-dev libmysofa-dev libnuma-dev libopenal-dev libopencore-amrnb-dev libopencore-amrwb-dev libopenjp2-7-dev libopenmpt-dev libopus-dev libopus-dev libpulse-dev librsvg2-dev librtmp-dev librubberband-dev libsdl2-dev libsdl2-dev libshine-dev libsmbclient-dev libsnappy-dev libsoxr-dev libspeex-dev libssh-dev libtesseract-dev libtheora-dev libtool libtwolame-dev libv4l-dev libva-dev libva-dev libvdpau-dev libvdpau-dev libvo-amrwbenc-dev libvorbis-dev libvorbis-dev libvorbis-dev libvpx-dev libvpx-dev libwavpack-dev libwebp-dev libx264-dev libx264-dev libx265-dev libx265-dev libxcb-shm0-dev libxcb-shm0-dev libxcb-xfixes0-dev libxcb-xfixes0-dev libxcb1-dev libxcb1-dev libxml2-dev libxvidcore-dev libzmq3-dev libzvbi-dev nvidia-cuda-toolkit nvidia-driver-440 opencl-dev p11-kit pkg-config pkg-config python3-dev python3-pip python3-setuptools python3-wheel python3.8 texinfo texinfo wget wget yasm zlib1g-dev zlib1g-dev - -# Install NASM -RUN curl -vfsSLO https://www.nasm.us/pub/nasm/releasebuilds/$NASM_VERSION/nasm-$NASM_VERSION.tar.bz2 \ - && tar -xjf nasm-$NASM_VERSION.tar.bz2 \ - && cd nasm-$NASM_VERSION \ - && ./autogen.sh \ - && ./configure \ - && make -j$(nproc) \ - && make install - -# Compile FFmpeg with CUDA support - -RUN git clone --recurse-submodules -b n$NVCODEC_VERSION --depth 1 https://git.videolan.org/git/ffmpeg/nv-codec-headers \ - && cd nv-codec-headers \ - && make install - -RUN curl -vfsSLO https://ffmpeg.org/releases/ffmpeg-$FFMPEG_VERSION.tar.bz2 \ - && tar -xjf ffmpeg-$FFMPEG_VERSION.tar.bz2 \ - && cd ffmpeg-$FFMPEG_VERSION \ - && ./configure --enable-cuda-sdk --enable-cuvid --enable-nonfree --enable-libnpp --enable-nvenc \ - --enable-gpl --enable-version3 \ - --enable-small --enable-avisynth --enable-chromaprint \ - --enable-frei0r --enable-gmp --enable-gnutls --enable-ladspa \ - --enable-libass --enable-libcaca --enable-libcdio \ - --enable-libcodec2 --enable-libfontconfig --enable-libfreetype \ - --enable-libfribidi --enable-libgme --enable-libgsm --enable-libjack \ - --enable-libmodplug --enable-libmp3lame --enable-libopencore-amrnb \ - --enable-libopencore-amrwb --enable-libopencore-amrwb \ - --enable-libopenjpeg --enable-libopenmpt --enable-libopus --enable-libpulse \ - --enable-librsvg --enable-librubberband --enable-librtmp --enable-libshine \ - --enable-libsnappy --enable-libsoxr --enable-libspeex \ - --enable-libssh --enable-libtesseract --enable-libtheora \ - --enable-libtwolame --enable-libv4l2 --enable-libvo-amrwbenc \ - --enable-libvorbis --enable-libvpx --enable-libwavpack --enable-libwebp \ - --enable-libx264 --enable-libx265 --enable-libxvid --enable-libxml2 \ - --enable-libzmq --enable-libzvbi --enable-lv2 \ - --enable-libmysofa \ - --enable-openal --enable-opencl --enable-opengl --enable-libdrm \ - --enable-libfdk-aac --enable-libbluray \ - && make -j$(nproc) \ - && make install - -# Add Nvidia's machine-learning repository for libcudnn7 and libcudnn7-dev -RUN curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub | apt-key add - &&\ - echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list - -# Install Video2X -RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.7 1 && update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.8 2 -RUN cd / && python3.8 -m pip install --upgrade pip &&\ - git clone --recurse-submodules --progress https://github.com/k4yt3x/video2x.git --depth=1 &&\ - python3.8 -m pip install -U -r video2x/src/requirements.txt - -# Compile drivers - -# example: docker build -t video2x . --build-arg driver=waifu2x_ncnn_vulkan -ARG driver=all - -# Check if driver exists. -SHELL ["/bin/bash", "-c"] -RUN drivers=(all waifu2x_caffe waifu2x_converter waifu2x_ncnn_vulkan) &&\ - case ${drivers[@]} in (*${driver,,}*) true ;; (*)\ - echo "ERROR: Unrecognized driver." >&2 &&\ - printf "%s " "Choices are: ${drivers[@]}" >&2 | printf "\n" >&2 &&\ - exit 1 ;;\ - esac - -RUN if [ "$driver" = "all" ] || [ "$driver" = "waifu2x_caffe" ] ; then \ - # nagadomi/caffe prerequisites - apt-fast update &&\ - apt-fast install -y --no-install-recommends build-essential cmake libboost-system-dev libboost-thread-dev libboost-filesystem-dev libboost-chrono-dev libboost-date-time-dev libboost-atomic-dev libboost-python-dev libprotobuf-dev protobuf-compiler libhdf5-dev liblmdb-dev libleveldb-dev libsnappy-dev libopencv-dev libatlas-base-dev python-numpy libgflags-dev libgoogle-glog-dev &&\ - # nagadomi/waifu2x-caffee-ubuntu prerequisites - apt-fast install -y --no-install-recommends libboost-iostreams-dev;\ - fi -# build waifu2x-caffe && install caffe -RUN if [ "$driver" = "all" ] || [ "$driver" = "waifu2x_caffe" ] ; then \ - git clone --recurse-submodules --depth=1 --progress --recurse-submodules https://github.com/nagadomi/waifu2x-caffe-ubuntu.git &&\ - cd waifu2x-caffe-ubuntu &&\ - git clone --recurse-submodules --progress --depth=1 https://github.com/nagadomi/caffe.git;\ - fi - -RUN if [ "$driver" = "all" ] || [ "$driver" = "waifu2x_caffe" ] ; then \ - apt-fast install --no-install-recommends -y gcc-8 libcudnn7 libcudnn7-dev &&\ - apt-get remove -y gcc g++ &&\ - ln -s /usr/bin/gcc-8 /usr/bin/gcc && ln -s /usr/bin/g++-8 /usr/bin/g++ &&\ - cd /waifu2x-caffe-ubuntu &&\ - mkdir build &&\ - cd build &&\ - cmake .. -DCMAKE_INSTALL_PREFIX=/usr &&\ - make -j$(nproc) install;\ - fi - -RUN if [ "$driver" = "all" ] || [ "$driver" = "waifu2x_caffe" ] ; then \ - # install waifu2x-caffe - cd /waifu2x-caffe-ubuntu/build &&\ - cp waifu2x-caffe ../bin/ &&\ - mv ../bin tempname &&\ - mv tempname /video2x/ &&\ - mv /video2x/tempname /video2x/waifu2x-caffe &&\ - rm -rf ../waifu2x-caffe-ubuntu ;\ - fi - -RUN if [ "$driver" = "all" ] || [ "$driver" = "waifu2x_ncnn_vulkan" ] ; then \ - apt-fast install -y --no-install-recommends software-properties-common build-essential cmake libvulkan-dev glslang-tools libprotobuf-dev protobuf-compiler &&\ - git clone --recurse-submodules --depth=1 --progress https://github.com/Tencent/ncnn.git &&\ - cd ncnn &&\ - mkdir -p build &&\ - cd build &&\ - cmake -DCMAKE_INSTALL_PREFIX=/usr -DNCNN_VULKAN=ON .. &&\ - make -j$(nproc) install &&\ - rm -rf ../../ncnn &&\ - cd / &&\ - # Compile waifu2x-ncnn-vulkan - git clone --recurse-submodules --depth=1 --progress https://github.com/nihui/waifu2x-ncnn-vulkan.git &&\ - cd waifu2x-ncnn-vulkan &&\ - mkdir -p build &&\ - cd build &&\ - cmake ../src &&\ - make -j$(nproc) &&\ - # Incall waifu2x-ncnn-vulkan - cd /waifu2x-ncnn-vulkan &&\ - mkdir waifu2x-ncnn-vulkan &&\ - mv models/models-cunet waifu2x-ncnn-vulkan/ &&\ - mv build/waifu2x-ncnn-vulkan waifu2x-ncnn-vulkan/ &&\ - mv waifu2x-ncnn-vulkan /video2x/ &&\ - rm -rf ../waifu2x-ncnn-vulkan ;\ - fi - -RUN if [ "$driver" = "all" ] || [ "$driver" = "waifu2x_converter" ] ;\ - then \ - # Prerequisits for waifu2x-converter-cpp - apt-fast install -y --no-install-recommends build-essential cmake libopencv-dev ocl-icd-opencl-dev &&\ - # Compile & Install - git clone --recurse-submodules --depth=1 --progress https://github.com/DeadSix27/waifu2x-converter-cpp &&\ - cd waifu2x-converter-cpp &&\ - mkdir build &&\ - cd build &&\ - cmake .. &&\ - make -j$(nproc) &&\ - #ldconfig &&\ - cp -r ../models_rgb ./&&\ - cd / ;\ - fi - -# Go! -#COPY entrypoint.sh / -#ENTRYPOINT ["/entrypoint.sh"] - -WORKDIR /host -ENTRYPOINT ["python3", "/video2x/src/video2x.py"] +WORKDIR /video2x/src +ENTRYPOINT ["python3.8", "/video2x/src/video2x.py"] ENV NVIDIA_DRIVER_CAPABILITIES all -# Docker image can ask questions now, if needed. ENV DEBIAN_FRONTEND teletype