From c687413bb8a07eb1beb30cb2fff1ea8265fc2b38 Mon Sep 17 00:00:00 2001 From: Jeremi Piotrowski Date: Mon, 26 Feb 2024 15:12:57 +0000 Subject: [PATCH] x11-drivers/nvidia-drivers: Enable persistence mode by default nvidia-persistenced keeps a reference to the GPU open at all times, which allows applications/containerd accessing gpus to start faster. Signed-off-by: Jeremi Piotrowski --- .../x11-drivers/nvidia-drivers/files/bin/setup-nvidia | 6 ++++++ ...104.05-r1.ebuild => nvidia-drivers-535.104.05-r2.ebuild} | 0 2 files changed, 6 insertions(+) rename sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/{nvidia-drivers-535.104.05-r1.ebuild => nvidia-drivers-535.104.05-r2.ebuild} (100%) diff --git a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia index 6ecb16d334..c694e40f03 100644 --- a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia +++ b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/files/bin/setup-nvidia @@ -9,6 +9,8 @@ set -euo pipefail NVIDIA_DOWNLOAD_BASEURL="https://us.download.nvidia.com/${NVIDIA_PRODUCT_TYPE}/" NVIDIA_DRIVER_BASENAME="NVIDIA-Linux-x86_64-${NVIDIA_DRIVER_VERSION}" NVIDIA_WORKDIR='nvidia-workdir' +# User can disable +: ${NVIDIA_PERSISTENCE_MODE:=on} FLATCAR_DEVELOPER_CONTAINER="flatcar_developer_container-${FLATCAR_RELEASE_VERSION}.bin" FLATCAR_DEVELOPER_CONTAINER_URL="" @@ -128,6 +130,10 @@ function install_and_load() { D=`grep nvidia-uvm /proc/devices | awk '{print $1}'` mknod -m 666 /dev/nvidia-uvm c $D 0 fi + + if [ "${NVIDIA_PERSISTENCE_MODE}" = "on" ]; then + nvidia-persistenced + fi } function verify_installation() { diff --git a/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/nvidia-drivers-535.104.05-r1.ebuild b/sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/nvidia-drivers-535.104.05-r2.ebuild similarity index 100% rename from sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/nvidia-drivers-535.104.05-r1.ebuild rename to sdk_container/src/third_party/coreos-overlay/x11-drivers/nvidia-drivers/nvidia-drivers-535.104.05-r2.ebuild