add nvenc support for gstreamer.
This commit is contained in:
parent
3c17dbe282
commit
bdff0841ec
@ -1,5 +1,55 @@
|
|||||||
ARG UBUNTU_RELEASE=20.04
|
ARG UBUNTU_RELEASE=20.04
|
||||||
ARG CUDA_VERSION=11.2.2
|
ARG CUDA_VERSION=11.4.3
|
||||||
|
ARG VIRTUALGL_VERSION=3.1
|
||||||
|
ARG GSTREAMER_VERSION=1.20
|
||||||
|
|
||||||
|
#
|
||||||
|
# STAGE 0: Build gstreamer with nvidia plugins.
|
||||||
|
#
|
||||||
|
FROM ubuntu:${UBUNTU_RELEASE} AS gstreamer
|
||||||
|
ARG GSTREAMER_VERSION
|
||||||
|
|
||||||
|
#
|
||||||
|
# install dependencies
|
||||||
|
ENV DEBIAN_FRONTEND=noninteractive
|
||||||
|
RUN set -eux; \
|
||||||
|
apt-get update; \
|
||||||
|
apt-get install -y --no-install-recommends \
|
||||||
|
# Install essentials
|
||||||
|
curl build-essential ca-certificates git \
|
||||||
|
# Install pip and ninja
|
||||||
|
python3-pip python-gi-dev ninja-build \
|
||||||
|
# Install build deps
|
||||||
|
autopoint autoconf automake autotools-dev libtool gettext bison flex gtk-doc-tools \
|
||||||
|
# Install libraries
|
||||||
|
libtool-bin \
|
||||||
|
libgtk2.0-dev \
|
||||||
|
libgl1-mesa-dev \
|
||||||
|
libopus-dev \
|
||||||
|
libpulse-dev \
|
||||||
|
libssl-dev \
|
||||||
|
libx264-dev \
|
||||||
|
libvpx-dev; \
|
||||||
|
# Install meson
|
||||||
|
pip3 install meson; \
|
||||||
|
#
|
||||||
|
# clean up
|
||||||
|
apt-get clean -y; \
|
||||||
|
rm -rf /var/lib/apt/lists/* /var/cache/apt/*
|
||||||
|
|
||||||
|
#
|
||||||
|
# build gstreamer
|
||||||
|
RUN set -eux; \
|
||||||
|
git clone --depth 1 --branch $GSTREAMER_VERSION https://gitlab.freedesktop.org/gstreamer/gstreamer.git /gstreamer/src; \
|
||||||
|
cd /gstreamer/src; \
|
||||||
|
mkdir -p /opt/gstreamer; \
|
||||||
|
meson --prefix /opt/gstreamer \
|
||||||
|
-Dgpl=enabled \
|
||||||
|
-Dugly=enabled \
|
||||||
|
-Dgst-plugins-ugly:x264=enabled \
|
||||||
|
build; \
|
||||||
|
ninja -C build; \
|
||||||
|
meson install -C build;
|
||||||
|
|
||||||
#
|
#
|
||||||
# STAGE 1: SERVER
|
# STAGE 1: SERVER
|
||||||
@ -51,13 +101,13 @@ RUN npm run build
|
|||||||
#
|
#
|
||||||
# STAGE 3: RUNTIME
|
# STAGE 3: RUNTIME
|
||||||
#
|
#
|
||||||
FROM nvcr.io/nvidia/cudagl:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_RELEASE} as runtime
|
FROM nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_RELEASE} as runtime
|
||||||
|
|
||||||
ARG UBUNTU_RELEASE
|
ARG UBUNTU_RELEASE
|
||||||
ARG CUDA_VERSION
|
ARG VIRTUALGL_VERSION
|
||||||
|
|
||||||
# Make all NVIDIA GPUs visible by default
|
# Make all NVIDIA GPUs visible by default
|
||||||
ENV NVIDIA_VISIBLE_DEVICES=all
|
ENV NVIDIA_VISIBLE_DEVICES all
|
||||||
# All NVIDIA driver capabilities should preferably be used, check `NVIDIA_DRIVER_CAPABILITIES` inside the container if things do not work
|
# All NVIDIA driver capabilities should preferably be used, check `NVIDIA_DRIVER_CAPABILITIES` inside the container if things do not work
|
||||||
ENV NVIDIA_DRIVER_CAPABILITIES all
|
ENV NVIDIA_DRIVER_CAPABILITIES all
|
||||||
|
|
||||||
@ -75,20 +125,55 @@ ARG USERNAME=neko
|
|||||||
ARG USER_UID=1000
|
ARG USER_UID=1000
|
||||||
ARG USER_GID=$USER_UID
|
ARG USER_GID=$USER_UID
|
||||||
|
|
||||||
|
RUN set -eux; \
|
||||||
|
dpkg --add-architecture i386; \
|
||||||
|
apt-get update; \
|
||||||
|
apt-get install -y --no-install-recommends \
|
||||||
|
# opengl base: https://gitlab.com/nvidia/container-images/opengl/-/blob/ubuntu20.04/base/Dockerfile
|
||||||
|
libxau6 libxau6:i386 \
|
||||||
|
libxdmcp6 libxdmcp6:i386 \
|
||||||
|
libxcb1 libxcb1:i386 \
|
||||||
|
libxext6 libxext6:i386 \
|
||||||
|
libx11-6 libx11-6:i386 \
|
||||||
|
# opengl runtime: https://gitlab.com/nvidia/container-images/opengl/-/blob/ubuntu20.04/glvnd/runtime/Dockerfile
|
||||||
|
libglvnd0 libglvnd0:i386 \
|
||||||
|
libgl1 libgl1:i386 \
|
||||||
|
libglx0 libglx0:i386 \
|
||||||
|
libegl1 libegl1:i386 \
|
||||||
|
libgles2 libgles2:i386 \
|
||||||
|
# hardware accleration utilities
|
||||||
|
libglu1 libglu1:i386 \
|
||||||
|
libvulkan-dev libvulkan-dev:i386 \
|
||||||
|
mesa-utils mesa-utils-extra \
|
||||||
|
mesa-va-drivers mesa-vulkan-drivers \
|
||||||
|
vainfo vdpauinfo; \
|
||||||
|
#
|
||||||
|
# install vulkan-utils or vulkan-tools depending on ubuntu release
|
||||||
|
if [ "${UBUNTU_RELEASE}" = "18.04" ]; then \
|
||||||
|
apt-get install -y --no-install-recommends vulkan-utils; \
|
||||||
|
else \
|
||||||
|
apt-get install -y --no-install-recommends vulkan-tools; \
|
||||||
|
fi; \
|
||||||
|
#
|
||||||
|
# create symlink for libnvrtc.so (needed for cudaconvert)
|
||||||
|
find /usr/local/cuda/lib64/ -maxdepth 1 -type l -name "*libnvrtc.so.*" -exec sh -c 'ln -sf {} /usr/local/cuda/lib64/libnvrtc.so' \;; \
|
||||||
|
#
|
||||||
|
# clean up
|
||||||
|
apt-get clean -y; \
|
||||||
|
rm -rf /var/lib/apt/lists/* /var/cache/apt/*
|
||||||
|
|
||||||
|
#
|
||||||
|
# add cuda to ld path, for gstreamer cuda plugins
|
||||||
|
ENV LD_LIBRARY_PATH="/usr/lib/x86_64-linux-gnu:/usr/lib/i386-linux-gnu${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}:/usr/local/cuda/lib:/usr/local/cuda/lib64"
|
||||||
|
|
||||||
RUN set -eux; \
|
RUN set -eux; \
|
||||||
apt-get update; \
|
apt-get update; \
|
||||||
#
|
#
|
||||||
# install dependencies
|
# install dependencies
|
||||||
apt-get install -y --no-install-recommends wget ca-certificates supervisor; \
|
apt-get install -y --no-install-recommends wget ca-certificates supervisor; \
|
||||||
apt-get install -y --no-install-recommends pulseaudio dbus-x11 xserver-xorg-video-dummy; \
|
apt-get install -y --no-install-recommends pulseaudio dbus-x11 xserver-xorg-video-dummy; \
|
||||||
apt-get install -y --no-install-recommends libcairo2 libxcb1 libxrandr2 libxv1 libopus0 libvpx6; \
|
apt-get install -y --no-install-recommends libcairo2 libxcb1 libxrandr2 libxv1 libopus0 libvpx6 libx264-155; \
|
||||||
#
|
apt-get install -y --no-install-recommends libgtk-3-bin software-properties-common cabextract aptitude vim curl; \
|
||||||
# hardware acclerations utilities
|
|
||||||
apt-get install -y --no-install-recommends libgtk-3-bin mesa-utils mesa-utils-extra mesa-va-drivers mesa-vulkan-drivers libvulkan-dev libvulkan-dev:i386 vdpauinfo; \
|
|
||||||
#
|
|
||||||
# gst
|
|
||||||
apt-get install -y --no-install-recommends libgstreamer1.0-dev libgstreamer-plugins-base1.0-dev \
|
|
||||||
gstreamer1.0-plugins-base gstreamer1.0-plugins-good gstreamer1.0-plugins-bad gstreamer1.0-plugins-ugly gstreamer1.0-pulseaudio; \
|
|
||||||
#
|
#
|
||||||
# install fonts
|
# install fonts
|
||||||
apt-get install -y --no-install-recommends \
|
apt-get install -y --no-install-recommends \
|
||||||
@ -124,18 +209,18 @@ RUN set -eux; \
|
|||||||
rm -rf /var/lib/apt/lists/* /var/cache/apt/*
|
rm -rf /var/lib/apt/lists/* /var/cache/apt/*
|
||||||
|
|
||||||
#
|
#
|
||||||
# install and configure Vulkan manually
|
# configure EGL and Vulkan manually
|
||||||
RUN set -eux; \
|
RUN VULKAN_API_VERSION=$(dpkg -s libvulkan1 | grep -oP 'Version: [0-9|\.]+' | grep -oP '[0-9]+(\.[0-9]+)(\.[0-9]+)') && \
|
||||||
apt-get update; \
|
# Configure EGL manually
|
||||||
if [ "${UBUNTU_RELEASE}" = "18.04" ]; then apt-get install -y --no-install-recommends vulkan-utils; else apt-get install -y --no-install-recommends vulkan-tools; fi; \
|
mkdir -p /usr/share/glvnd/egl_vendor.d/ && \
|
||||||
#
|
echo "{\n\
|
||||||
# clean up
|
\"file_format_version\" : \"1.0.0\",\n\
|
||||||
apt-get clean -y; \
|
\"ICD\": {\n\
|
||||||
rm -rf /var/lib/apt/lists/* /var/cache/apt/*; \
|
\"library_path\": \"libEGL_nvidia.so.0\"\n\
|
||||||
#
|
}\n\
|
||||||
# configure vulkan
|
}" > /usr/share/glvnd/egl_vendor.d/10_nvidia.json && \
|
||||||
VULKAN_API_VERSION=$(dpkg -s libvulkan1 | grep -oP 'Version: [0-9|\.]+' | grep -oP '[0-9]+(\.[0-9]+)(\.[0-9]+)'); \
|
# Configure Vulkan manually
|
||||||
mkdir -p /etc/vulkan/icd.d/; \
|
mkdir -p /etc/vulkan/icd.d/ && \
|
||||||
echo "{\n\
|
echo "{\n\
|
||||||
\"file_format_version\" : \"1.0.0\",\n\
|
\"file_format_version\" : \"1.0.0\",\n\
|
||||||
\"ICD\": {\n\
|
\"ICD\": {\n\
|
||||||
@ -146,7 +231,6 @@ RUN set -eux; \
|
|||||||
|
|
||||||
#
|
#
|
||||||
# install VirtualGL and make libraries available for preload
|
# install VirtualGL and make libraries available for preload
|
||||||
ARG VIRTUALGL_VERSION=3.1
|
|
||||||
RUN set -eux; \
|
RUN set -eux; \
|
||||||
apt-get update; \
|
apt-get update; \
|
||||||
wget "https://sourceforge.net/projects/virtualgl/files/virtualgl_${VIRTUALGL_VERSION}_amd64.deb"; \
|
wget "https://sourceforge.net/projects/virtualgl/files/virtualgl_${VIRTUALGL_VERSION}_amd64.deb"; \
|
||||||
@ -181,6 +265,16 @@ ENV NEKO_PASSWORD=neko
|
|||||||
ENV NEKO_PASSWORD_ADMIN=admin
|
ENV NEKO_PASSWORD_ADMIN=admin
|
||||||
ENV NEKO_BIND=:8080
|
ENV NEKO_BIND=:8080
|
||||||
|
|
||||||
|
#
|
||||||
|
# set gstreamer envs
|
||||||
|
ENV PATH="/opt/gstreamer/bin:${PATH}"
|
||||||
|
ENV LD_LIBRARY_PATH="/opt/gstreamer/lib/x86_64-linux-gnu${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}"
|
||||||
|
ENV PKG_CONFIG_PATH="/opt/gstreamer/lib/x86_64-linux-gnu/pkgconfig${PKG_CONFIG_PATH:+:${PKG_CONFIG_PATH}}"
|
||||||
|
|
||||||
|
#
|
||||||
|
# copy gstreamer from previous stage
|
||||||
|
COPY --from=gstreamer /opt/gstreamer /opt/gstreamer
|
||||||
|
|
||||||
#
|
#
|
||||||
# copy static files from previous stages
|
# copy static files from previous stages
|
||||||
COPY --from=server /src/bin/neko /usr/bin/neko
|
COPY --from=server /src/bin/neko /usr/bin/neko
|
||||||
|
Reference in New Issue
Block a user