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 <jpiotrowski@microsoft.com>
This commit is contained in:
Jeremi Piotrowski 2024-02-26 15:12:57 +00:00
parent cb1ddeb5a5
commit c687413bb8
2 changed files with 6 additions and 0 deletions

View File

@ -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() {