Vgl-Egl-Xvfb.def

Bootstrap: docker
From: nvidia/cuda:12.2.2-cudnn8-devel-ubuntu22.04

%post -c /bin/bash

    # Enable error handling
    set -euxo pipefail
    
    # Set DEBIAN_FRONTEND to noninteractive for package installation
    export DEBIAN_FRONTEND=noninteractive

    # Update apt-get and install necessary packages
    apt-get update && apt-get install --no-install-recommends -y \
        software-properties-common \
        jq \
        curl \
        terminator \
        tmux \
        vim \
        libsm6 \
        libxext6 \
        libxrender-dev \
        gedit \
        git \
        python3-venv \
        python3-dev \
        python3-pip \
        openssh-client \
        unzip \
        htop \
        libopenni-dev \
        apt-utils \
        usbutils \
        dialog \
        ffmpeg \
        nvidia-settings \
        libffi-dev \
        flex \
        bison \
        build-essential \
        wget \
        pciutils \
        mesa-utils \
        xauth \
        xvfb \
        x11vnc

    apt-get remove -y python3-blinker


    # Download the Miniforge installer for the architecture
    curl -L -O "https://github.com/conda-forge/miniforge/releases/latest/download/Miniforge3-$(uname)-$(uname -m).sh"

    # Run the installer in silent mode (no user prompts)
    bash Miniforge3-$(uname)-$(uname -m).sh -b -p /opt/miniforge

    # Clean up installer to save space
    rm Miniforge3-$(uname)-$(uname -m).sh

    # Install mamba (you could also install other packages you need here)
    /opt/miniforge/bin/conda install -n base -c conda-forge mamba -y
    
    # Install VirtualGL for GPU-accelerated remote desktop applications
    VIRTUALGL_VERSION=$(curl -fsSL "https://api.github.com/repos/VirtualGL/virtualgl/releases/67016359" | jq -r '.tag_name' | sed 's/[^0-9\.\-]*//g') && \
    curl -fsSL -O https://sourceforge.net/projects/virtualgl/files/virtualgl_${VIRTUALGL_VERSION}_amd64.deb && \
    apt-get update && apt-get install -y --no-install-recommends ./virtualgl_${VIRTUALGL_VERSION}_amd64.deb && \
    rm virtualgl_${VIRTUALGL_VERSION}_amd64.deb && \
    rm -rf /var/lib/apt/lists/* && \
    chmod u+s /usr/lib/libvglfaker.so && \
    chmod u+s /usr/lib/libdlfaker.so && \

    # Clean up unnecessary files after installation
    apt-get clean
    rm -rf /var/lib/apt/lists/*

    # Update pip
    pip install --upgrade pip

%files
    # Copy the 10_nvidia.json file into the container
    10_nvidia.json /usr/share/glvnd/egl_vendor.d/10_nvidia.json

%environment
    # Set environment variables for the runtime
    export PATH=/opt/miniforge/bin:$PATH
    export LD_LIBRARY_PATH=/usr/lib/x86_64-linux-gnu/:$LD_LIBRARY_PATH
    export PULSE_SERVER=none
    export LIBGL_ALWAYS_INDIRECT=1
    export VGL_DISPLAY=egl
    export VGL_REFRESHRATE=60
    export DISPLAY=:0  # Set DISPLAY variable for Xvfb
    export PYTHONUNBUFFERED=1

%runscript
    # Ensure DISPLAY variable is set
    export DISPLAY=:0

    # Execute the provided command or default to bash
    if [ "$#" -eq 0 ]; then
        exec /bin/bash
    else
        exec "$@"
    fi