Install the NVIDIA patch on container restarts
This commit is contained in:
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user