Dockerfile

# syntax=docker/dockerfile:1.4
FROM rubensa/ubuntu-tini-dev
LABEL author="Ruben Suarez <rubensa@gmail.com>"

# Architecture component of TARGETPLATFORM (platform of the build result)
ARG TARGETARCH

# If defined, install specified NVIDIA driver version
ARG NVIDIA_VERSION

# Tell docker that all future commands should be run as root
USER root

# Set root home directory
ENV HOME=/root

# DBUS_SESSION_BUS_ADDRESS is lost with sudo (https://github.com/SeleniumHQ/docker-selenium/issues/358)
RUN printf "\nDefaults env_keep += \"DBUS_SESSION_BUS_ADDRESS\"\n" >> /etc/sudoers

# suppress GTK warnings about accessibility
# (WARNING **: Couldn't connect to accessibility bus: Failed to connect to socket /tmp/dbus-dw0fOAy4vj: Connection refused)
ENV NO_AT_BRIDGE 1

# Avoid warnings by switching to noninteractive
ENV DEBIAN_FRONTEND=noninteractive

# Configure apt
RUN apt-get update

# Install Google Noto font family
RUN <<EOT
echo "# Installing Google Noto font family..."
apt-get -y install fonts-noto 2>&1
EOT

# Install software and libraries needed to share X11 between host and container
RUN <<EOT
echo "# Installing kmod, libglib2.0-bin, libgl1, libglx-mesa0, libgl1-mesa-dri, pulseaudio-utils, cups-client, x11-utils, mesa-utils, mesa-utils-extra and va-driver-all..."
apt-get -y install --no-install-recommends kmod libglib2.0-bin libgl1 libglx-mesa0 libgl1-mesa-dri pulseaudio-utils cups-client x11-utils mesa-utils mesa-utils-extra va-driver-all 2>&1
EOT

# Configure user (add to audio and video groups)
RUN <<EOT
echo "# Configuring '${USER_NAME}' for X11 functionallity..."
#
# Assign audio group to non-root user
usermod -a -G audio ${USER_NAME}
#
# Assign video group to non-root user
usermod -a -G video ${USER_NAME}
EOT

# Install NVIDIA drivers in the image if NVIDIA_VERSION arg set
RUN <<EOT
if [ ! -z ${NVIDIA_VERSION} ] ; then
  if [ "$TARGETARCH" = "arm64" ]; then
    TARGET=aarch64
  elif [ "$TARGETARCH" = "amd64" ]; then
    TARGET=x86_64
  else
    TARGET=$TARGETARCH
  fi
  echo "# Downloading NVIDIA drivers matching host version (${NVIDIA_VERSION})..."
  curl -sSLO http://us.download.nvidia.com/XFree86/Linux-${TARGET}/${NVIDIA_VERSION}/NVIDIA-Linux-${TARGET}-${NVIDIA_VERSION}.run
  chmod +x NVIDIA-Linux-${TARGET}-${NVIDIA_VERSION}.run
  echo "# Installing NVIDIA drivers..."
  ./NVIDIA-Linux-${TARGET}-${NVIDIA_VERSION}.run --ui=none --no-kernel-module --no-install-compat32-libs --install-libglvnd --no-questions
  rm NVIDIA-Linux-${TARGET}-${NVIDIA_VERSION}.run
fi
EOT

# Add script to allow nvidia drivers installation on user interactive session
ADD install-nvidia-drivers.sh /usr/bin/install-nvidia-drivers.sh
RUN <<EOT
echo "# Configuring '${USER_NAME}' for NVIDIA drivers auto installation if NVIDIA_VERSION env variable is set..."
#
# Enable runtime nvidia drivers installation
chmod +x /usr/bin/install-nvidia-drivers.sh
#
# Configure nvidia drivers installation for the non-root user
printf "\n. /usr/bin/install-nvidia-drivers.sh\n" >> /home/${USER_NAME}/.bashrc
EOT

# Clean up apt
RUN <<EOT
apt-get autoremove -y
apt-get clean -y
rm -rf /var/lib/apt/lists/*
EOT

# Switch back to dialog for any ad-hoc use of apt-get
ENV DEBIAN_FRONTEND=

# Tell docker that all future commands should be run as the non-root user
USER ${USER_NAME}

# Set user home directory (see: https://github.com/microsoft/vscode-remote-release/issues/852)
ENV HOME=/home/$USER_NAME