Install the NVIDIA patch on container restarts

This commit is contained in:
Josh.5
2023-09-04 17:43:05 +12:00
parent 63137d43ec
commit c8d613da5b

View File

@@ -44,63 +44,62 @@ function download_driver {
function install_nvidia_driver { function install_nvidia_driver {
# Check here if the currently installed version matches using nvidia-settings # Check here if the currently installed version matches using nvidia-settings
nvidia_settings_version=$(nvidia-settings --version 2> /dev/null | grep version | cut -d ' ' -f 4) nvidia_settings_version=$(nvidia-settings --version 2> /dev/null | grep version | cut -d ' ' -f 4)
[ "${nvidia_settings_version:-}X" == "${nvidia_host_driver_version:-}X" ] && return 0; if [ "${nvidia_settings_version:-}X" != "${nvidia_host_driver_version:-}X" ]; then
# Download the driver (if it does not yet exist locally)
download_driver
# Download the driver (if it does not yet exist locally) # if command -v pacman &> /dev/null; then
download_driver # echo "Install NVIDIA vulkan utils" \
# && pacman -Syu --noconfirm --needed \
# if command -v pacman &> /dev/null; then # lib32-nvidia-utils \
# echo "Install NVIDIA vulkan utils" \ # lib32-vulkan-icd-loader
# && pacman -Syu --noconfirm --needed \ # nvidia-utils \
# lib32-nvidia-utils \ # vulkan-icd-loader \
# lib32-vulkan-icd-loader # && echo
# nvidia-utils \ # fi
# vulkan-icd-loader \ if (($(echo $nvidia_host_driver_version | cut -d '.' -f 1) > 500)); then
# && echo echo "Installing NVIDIA driver v${nvidia_host_driver_version:?} to match what is running on the host"
# fi chmod +x "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run"
"${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" \
if (($(echo $nvidia_host_driver_version | cut -d '.' -f 1) > 500)); then --silent \
echo "Installing NVIDIA driver v${nvidia_host_driver_version:?} to match what is running on the host" --accept-license \
chmod +x "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" --skip-depmod \
"${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" \ --skip-module-unload \
--silent \ --no-kernel-modules \
--accept-license \ --no-kernel-module-source \
--skip-depmod \ --install-compat32-libs \
--skip-module-unload \ --no-nouveau-check \
--no-kernel-modules \ --no-nvidia-modprobe \
--no-kernel-module-source \ --no-systemd \
--install-compat32-libs \ --no-distro-scripts \
--no-nouveau-check \ --no-rpms \
--no-nvidia-modprobe \ --no-backup \
--no-systemd \ --no-check-for-alternate-installs \
--no-distro-scripts \ --no-libglx-indirect \
--no-rpms \ --no-install-libglvnd \
--no-backup \ > "${USER_HOME:?}/Downloads/nvidia_gpu_install.log" 2>&1
--no-check-for-alternate-installs \ else
--no-libglx-indirect \ echo "Installing Legacy NVIDIA driver v${nvidia_host_driver_version:?} to match what is running on the host"
--no-install-libglvnd \ chmod +x "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run"
> "${USER_HOME:?}/Downloads/nvidia_gpu_install.log" 2>&1 "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" \
else --silent \
echo "Installing Legacy NVIDIA driver v${nvidia_host_driver_version:?} to match what is running on the host" --accept-license \
chmod +x "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" --skip-depmod \
"${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" \ --skip-module-unload \
--silent \ --no-kernel-module \
--accept-license \ --no-kernel-module-source \
--skip-depmod \ --install-compat32-libs \
--skip-module-unload \ --no-nouveau-check \
--no-kernel-module \ --no-nvidia-modprobe \
--no-kernel-module-source \ --no-systemd \
--install-compat32-libs \ --no-distro-scripts \
--no-nouveau-check \ --no-rpms \
--no-nvidia-modprobe \ --no-backup \
--no-systemd \ --no-check-for-alternate-installs \
--no-distro-scripts \ --no-libglx-indirect \
--no-rpms \ --no-install-libglvnd \
--no-backup \ > "${USER_HOME:?}/Downloads/nvidia_gpu_install.log" 2>&1
--no-check-for-alternate-installs \ fi
--no-libglx-indirect \
--no-install-libglvnd \
> "${USER_HOME:?}/Downloads/nvidia_gpu_install.log" 2>&1
fi fi
# REF: https://github.com/keylase/nvidia-patch#docker-support # REF: https://github.com/keylase/nvidia-patch#docker-support
if [ "${NVIDIA_PATCH_VERSION:-}X" != "X" ]; then if [ "${NVIDIA_PATCH_VERSION:-}X" != "X" ]; then