3cce8f79f8
The GitHub servers had a small outage today while I was working. This prevented the containers from being able to start. This change allows for the patch to be re-used between container re-creations and also prevents the failure from stopping boot. If the patch fails to download or install, then it will continue with an unpatched driver.
214 lines
9.3 KiB
Bash
214 lines
9.3 KiB
Bash
|
|
# Fech NVIDIA GPU device (if one exists)
|
|
if [ "${NVIDIA_VISIBLE_DEVICES:-}" = "all" ]; then
|
|
export gpu_select=$(nvidia-smi --format=csv --query-gpu=uuid 2> /dev/null | sed -n 2p)
|
|
elif [ "${NVIDIA_VISIBLE_DEVICES:-}X" = "X" ]; then
|
|
export gpu_select=$(nvidia-smi --format=csv --query-gpu=uuid 2> /dev/null | sed -n 2p)
|
|
else
|
|
export gpu_select=$(nvidia-smi --format=csv --id=$(echo "$NVIDIA_VISIBLE_DEVICES" | cut -d ',' -f1) --query-gpu=uuid | sed -n 2p)
|
|
if [ "${gpu_select:-}X" = "X" ]; then
|
|
export gpu_select=$(nvidia-smi --format=csv --query-gpu=uuid 2> /dev/null | sed -n 2p)
|
|
fi
|
|
fi
|
|
|
|
# NVIDIA Params
|
|
export nvidia_pci_address="$(nvidia-smi --format=csv --query-gpu=pci.bus_id --id="${gpu_select:?}" 2> /dev/null | sed -n 2p | cut -d ':' -f2,3)"
|
|
export nvidia_gpu_name=$(nvidia-smi --format=csv --query-gpu=name --id="${gpu_select:?}" 2> /dev/null | sed -n 2p)
|
|
export nvidia_host_driver_version="$(nvidia-smi 2> /dev/null | grep NVIDIA-SMI | cut -d ' ' -f3)"
|
|
|
|
# Intel params
|
|
# This figures out if it's an intel CPU with integrated GPU
|
|
export intel_cpu_model="$(lscpu | grep 'Model name:' | grep -i intel | cut -d':' -f2 | xargs)"
|
|
# We need to check if the user has an intel ARC GPU as well
|
|
export intel_gpu_model="$(lspci | grep -i "VGA compatible controller: Intel" | cut -d':' -f3 | xargs)"
|
|
|
|
# AMD params
|
|
export amd_cpu_model="$(lscpu | grep 'Model name:' | grep -i amd | cut -d':' -f2 | xargs)"
|
|
export amd_gpu_model="$(lspci | grep -i vga | grep -i amd)"
|
|
|
|
function download_driver {
|
|
mkdir -p "${USER_HOME:?}/Downloads"
|
|
chown -R ${USER:?} "${USER_HOME:?}/Downloads"
|
|
|
|
if [[ ! -f "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" ]]; then
|
|
echo "Downloading driver v${nvidia_host_driver_version:?}"
|
|
wget -q --show-progress --progress=bar:force:noscroll \
|
|
-O /tmp/NVIDIA.run \
|
|
"http://download.nvidia.com/XFree86/Linux-x86_64/${nvidia_host_driver_version:?}/NVIDIA-Linux-x86_64-${nvidia_host_driver_version:?}.run"
|
|
[[ $? -gt 0 ]] && echo "Error downloading driver. Exit!" && return 1
|
|
|
|
mv /tmp/NVIDIA.run "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run"
|
|
fi
|
|
}
|
|
|
|
function install_nvidia_driver {
|
|
# 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)
|
|
if [ "${nvidia_settings_version:-}X" != "${nvidia_host_driver_version:-}X" ]; then
|
|
# Download the driver (if it does not yet exist locally)
|
|
download_driver
|
|
|
|
# if command -v pacman &> /dev/null; then
|
|
# echo "Install NVIDIA vulkan utils" \
|
|
# && pacman -Syu --noconfirm --needed \
|
|
# lib32-nvidia-utils \
|
|
# lib32-vulkan-icd-loader
|
|
# nvidia-utils \
|
|
# vulkan-icd-loader \
|
|
# && echo
|
|
# fi
|
|
if (($(echo $nvidia_host_driver_version | cut -d '.' -f 1) > 500)); then
|
|
echo "Installing NVIDIA driver v${nvidia_host_driver_version:?} to match what is running on the host"
|
|
chmod +x "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run"
|
|
"${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" \
|
|
--silent \
|
|
--accept-license \
|
|
--skip-depmod \
|
|
--skip-module-unload \
|
|
--no-kernel-modules \
|
|
--no-kernel-module-source \
|
|
--install-compat32-libs \
|
|
--no-nouveau-check \
|
|
--no-nvidia-modprobe \
|
|
--no-systemd \
|
|
--no-distro-scripts \
|
|
--no-rpms \
|
|
--no-backup \
|
|
--no-check-for-alternate-installs \
|
|
--no-libglx-indirect \
|
|
--no-install-libglvnd \
|
|
> "${USER_HOME:?}/Downloads/nvidia_gpu_install.log" 2>&1
|
|
else
|
|
echo "Installing Legacy NVIDIA driver v${nvidia_host_driver_version:?} to match what is running on the host"
|
|
chmod +x "${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run"
|
|
"${USER_HOME:?}/Downloads/NVIDIA_${nvidia_host_driver_version:?}.run" \
|
|
--silent \
|
|
--accept-license \
|
|
--skip-depmod \
|
|
--skip-module-unload \
|
|
--no-kernel-module \
|
|
--no-kernel-module-source \
|
|
--install-compat32-libs \
|
|
--no-nouveau-check \
|
|
--no-nvidia-modprobe \
|
|
--no-systemd \
|
|
--no-distro-scripts \
|
|
--no-rpms \
|
|
--no-backup \
|
|
--no-check-for-alternate-installs \
|
|
--no-libglx-indirect \
|
|
--no-install-libglvnd \
|
|
> "${USER_HOME:?}/Downloads/nvidia_gpu_install.log" 2>&1
|
|
fi
|
|
fi
|
|
}
|
|
|
|
function patch_nvidia_driver {
|
|
# REF: https://github.com/keylase/nvidia-patch#docker-support
|
|
if [ "${NVIDIA_PATCH_VERSION:-}X" != "X" ]; then
|
|
# Don't fail boot if something goes wrong here. Set +e
|
|
(
|
|
set +e
|
|
if [ ! -f "${USER_HOME:?}/Downloads/nvidia-patch.${NVIDIA_PATCH_VERSION:?}.sh" ]; then
|
|
echo "Fetch NVIDIA NVENC patch"
|
|
wget -q --show-progress --progress=bar:force:noscroll \
|
|
-O "${USER_HOME:?}/Downloads/nvidia-patch.${NVIDIA_PATCH_VERSION:?}.sh" \
|
|
"https://raw.githubusercontent.com/keylase/nvidia-patch/${NVIDIA_PATCH_VERSION:?}/patch.sh"
|
|
fi
|
|
if [ ! -f "${USER_HOME:?}/Downloads/nvidia-patch-fbc.${NVIDIA_PATCH_VERSION:?}.sh" ]; then
|
|
echo "Fetch NVIDIA NvFBC patch"
|
|
wget -q --show-progress --progress=bar:force:noscroll \
|
|
-O "${USER_HOME:?}/Downloads/nvidia-patch-fbc.${NVIDIA_PATCH_VERSION:?}.sh" \
|
|
"https://raw.githubusercontent.com/keylase/nvidia-patch/${NVIDIA_PATCH_VERSION:?}/patch-fbc.sh"
|
|
fi
|
|
chmod +x \
|
|
"${USER_HOME:?}/Downloads/nvidia-patch.${NVIDIA_PATCH_VERSION:?}.sh" \
|
|
"${USER_HOME:?}/Downloads/nvidia-patch-fbc.${NVIDIA_PATCH_VERSION:?}.sh"
|
|
|
|
echo "Install NVIDIA driver patches"
|
|
echo "/patched-lib" > /etc/ld.so.conf.d/000-patched-lib.conf
|
|
mkdir -p "/patched-lib"
|
|
PATCH_OUTPUT_DIR="/patched-lib" "${USER_HOME:?}/Downloads/nvidia-patch.${NVIDIA_PATCH_VERSION:?}.sh"
|
|
PATCH_OUTPUT_DIR="/patched-lib" "${USER_HOME:?}/Downloads/nvidia-patch-fbc.${NVIDIA_PATCH_VERSION:?}.sh"
|
|
|
|
pushd "/patched-lib" &> /dev/null || { echo "Error: Failed to push directory to /patched-lib"; exit 1; }
|
|
for f in * ; do
|
|
suffix="${f##*.so}"
|
|
name="$(basename "$f" "$suffix")"
|
|
[ -h "$name" ] || ln -sf "$f" "$name"
|
|
[ -h "$name" ] || ln -sf "$f" "$name.1"
|
|
done
|
|
ldconfig
|
|
popd &> /dev/null || { echo "Error: Failed to pop directory out of /patched-lib"; exit 1; }
|
|
)
|
|
else
|
|
echo "Leaving NVIDIA driver stock without patching"
|
|
fi
|
|
}
|
|
|
|
function install_amd_gpu_driver {
|
|
echo "Install AMD vulkan driver"
|
|
if command -v pacman &> /dev/null; then
|
|
pacman -Syu --noconfirm --needed \
|
|
lib32-vulkan-icd-loader \
|
|
lib32-vulkan-radeon \
|
|
vulkan-icd-loader \
|
|
vulkan-radeon
|
|
# There is currently nothing to install inside the debian container. This already comes with the vulken drives that are required
|
|
# elif command -v apt-get &> /dev/null; then
|
|
# [[ "${APT_UPDATED:-false}" == 'false' ]] && apt-get update && export APT_UPDATED=true
|
|
# apt-get install -y \
|
|
# libvulkan1 \
|
|
# libvulkan1:i386 \
|
|
# mesa-vulkan-drivers \
|
|
# mesa-vulkan-drivers:i386
|
|
fi
|
|
}
|
|
|
|
function install_intel_gpu_driver {
|
|
echo "Install Intel vulkan driver"
|
|
if command -v pacman &> /dev/null; then
|
|
pacman -Syu --noconfirm --needed \
|
|
lib32-vulkan-icd-loader \
|
|
lib32-vulkan-intel \
|
|
vulkan-icd-loader \
|
|
vulkan-intel
|
|
# There is currently nothing to install inside the debian container. This already comes with the vulken drives that are required
|
|
# elif command -v apt-get &> /dev/null; then
|
|
# [[ "${APT_UPDATED:-false}" == 'false' ]] && apt-get update && export APT_UPDATED=true
|
|
# apt-get install -y \
|
|
# libvulkan1 \
|
|
# libvulkan1:i386 \
|
|
# mesa-vulkan-drivers \
|
|
# mesa-vulkan-drivers:i386
|
|
fi
|
|
}
|
|
|
|
# Intel Arc GPU or Intel CPU with possible iGPU
|
|
if [ "${intel_gpu_model:-}X" != "X" ]; then
|
|
echo "**** Found Intel device '${intel_gpu_model:?}' ****"
|
|
install_intel_gpu_driver
|
|
elif [ "${intel_cpu_model:-}X" != "X" ]; then
|
|
echo "**** Found Intel device '${intel_cpu_model:?}' ****"
|
|
install_intel_gpu_driver
|
|
else
|
|
echo "**** No Intel device found ****"
|
|
fi
|
|
# AMD GPU
|
|
if [ "${amd_gpu_model:-}X" != "X" ]; then
|
|
echo "**** Found AMD device '${amd_gpu_model:?}' ****"
|
|
install_amd_gpu_driver
|
|
else
|
|
echo "**** No AMD device found ****"
|
|
fi
|
|
# NVIDIA GPU
|
|
if [ "${nvidia_pci_address:-}X" != "X" ]; then
|
|
echo "**** Found NVIDIA device '${nvidia_gpu_name:?}' ****"
|
|
install_nvidia_driver
|
|
patch_nvidia_driver
|
|
else
|
|
echo "**** No NVIDIA device found ****"
|
|
fi
|
|
|
|
echo "DONE"
|