Containerfile

ARG BASE_IMAGE="${BASE_IMAGE}"
ARG FEDORA_VERSION="${FEDORA_VERSION}"
ARG NVIDIA_VERSION="${NVIDIA_VERSION}"
ARG USE_LTS_KERNEL="${USE_LTS_KERNEL}"

# Build NVIDIA drivers and other source packages

FROM ${BASE_IMAGE}:${FEDORA_VERSION} as builder
ARG NVIDIA_VERSION="${NVIDIA_VERSION}"
ARG USE_LTS_KERNEL="${USE_LTS_KERNEL}"
ARG BUILDROOT=/build
COPY build_files/ "${BUILDROOT}"
RUN rpm-ostree cliwrap install-to-root / \
  && "${BUILDROOT}"/scripts/build-nvidia-modules.sh

# Build final image

FROM ${BASE_IMAGE}:${FEDORA_VERSION}
ARG NVIDIA_VERSION="${NVIDIA_VERSION}"
ARG USE_LTS_KERNEL="${USE_LTS_KERNEL}"
ARG BUILDROOT=/build

# Copy build scripts
COPY build_files/ "${BUILDROOT}"
# Copy built modules from builder step
COPY --from=builder /tmp/built /tmp/builder
RUN ls -la /tmp/builder/nvidia
# Copy configuration files to root
COPY ./system_files/. /
# Run installer and commit image
RUN rpm-ostree cliwrap install-to-root / \
  && "${BUILDROOT}"/scripts/install.sh \
  && ostree container commit \
  && rm -rf "${BUILDROOT}" \
  && mkdir -p /var/tmp && chmod -R 1777 /var/tmp