mirror of
https://github.com/kuhyx/scripts.git
synced 2026-07-04 13:03:05 +02:00
70 lines
3.5 KiB
Bash
Executable File
70 lines
3.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# NVIDIA driver selection & installation (split from detect script)
|
|
# Expects GPU_VENDOR=nvidia
|
|
# Outputs: NVIDIA_DRIVER_PACKAGE
|
|
set -e
|
|
|
|
[ "$GPU_VENDOR" = "nvidia" ] || { echo "NVIDIA installer invoked but GPU_VENDOR=$GPU_VENDOR"; exit 0; }
|
|
|
|
_build_aur_pkg() {
|
|
local pkg="$1"; local repo_url="https://aur.archlinux.org/${pkg}.git";
|
|
mkdir -p "$HOME/aur"; cd "$HOME/aur";
|
|
if [ ! -d "$pkg" ]; then git clone "$repo_url"; else (cd "$pkg" && git fetch -q --all && git reset -q --hard origin/HEAD || git pull --ff-only || true); fi
|
|
cd "$pkg"; rm -f -- *.pkg.tar.* 2>/dev/null || true
|
|
yes | makepkg -s -c -C --noconfirm --needed || return 1
|
|
local built=( *.pkg.tar.zst ); yes | sudo pacman -U --noconfirm "${built[@]}"
|
|
}
|
|
|
|
_choose_nvidia_pkg() {
|
|
local have_linux have_linux_lts multiple_kernels driver_pkg prefer_open
|
|
prefer_open=${NVIDIA_PREFER_OPEN:-1}
|
|
pacman -Qq | grep -qx linux && have_linux=1 || have_linux=0
|
|
pacman -Qq | grep -qx linux-lts && have_linux_lts=1 || have_linux_lts=0
|
|
if [ $((have_linux + have_linux_lts)) -gt 1 ]; then multiple_kernels=1; else multiple_kernels=0; fi
|
|
if ! command -v nvidia-detect >/dev/null 2>&1; then yes | sudo pacman -Sy --noconfirm nvidia-detect || true; fi
|
|
local detect_out="$(nvidia-detect 2>/dev/null || true)"
|
|
if echo "$detect_out" | grep -q '470'; then driver_pkg='nvidia-470xx-dkms';
|
|
elif echo "$detect_out" | grep -q '390'; then driver_pkg='nvidia-390xx-dkms';
|
|
elif echo "$detect_out" | grep -q '340'; then driver_pkg='nvidia-340xx-dkms';
|
|
else
|
|
if [ "$multiple_kernels" = 1 ]; then
|
|
if [ "$prefer_open" = 1 ] && pacman -Si nvidia-open-dkms >/dev/null 2>&1; then driver_pkg='nvidia-open-dkms'; else driver_pkg='nvidia-dkms'; fi
|
|
else
|
|
if [ "$have_linux_lts" = 1 ] && [ "$have_linux" = 0 ]; then
|
|
if [ "$prefer_open" = 1 ] && pacman -Si nvidia-open-lts >/dev/null 2>&1; then driver_pkg='nvidia-open-lts'; else driver_pkg='nvidia-lts'; fi
|
|
else
|
|
if [ "$prefer_open" = 1 ] && pacman -Si nvidia-open >/dev/null 2>&1; then driver_pkg='nvidia-open'; else driver_pkg='nvidia'; fi
|
|
fi
|
|
fi
|
|
fi
|
|
echo "$driver_pkg"
|
|
}
|
|
|
|
_remove_conflicting_nvidia_pkgs() {
|
|
local keep="$1"; local candidates=(nvidia nvidia-lts nvidia-dkms nvidia-open nvidia-open-lts nvidia-open-dkms nvidia-470xx-dkms nvidia-390xx-dkms nvidia-340xx-dkms)
|
|
local to_remove=()
|
|
for p in "${candidates[@]}"; do
|
|
if pacman -Qi "$p" >/dev/null 2>&1 && [ "$p" != "$keep" ]; then to_remove+=("$p"); fi
|
|
done
|
|
if [ ${#to_remove[@]} -gt 0 ]; then yes | sudo pacman -Rns --noconfirm "${to_remove[@]}" || true; fi
|
|
}
|
|
|
|
_install_nvidia_stack() {
|
|
local driver_pkg="$1"
|
|
if [[ "$driver_pkg" == nvidia-*xx-dkms ]]; then _build_aur_pkg "$driver_pkg"; else yes | sudo pacman -Sy --noconfirm "$driver_pkg"; fi
|
|
local utils_pkg="nvidia-utils" utils32_pkg="lib32-nvidia-utils"
|
|
if ! pacman -Qi "$utils_pkg" >/dev/null 2>&1; then yes | sudo pacman -Sy --noconfirm "$utils_pkg"; fi
|
|
if grep -q '^\[multilib\]' /etc/pacman.conf; then
|
|
if ! pacman -Qi "$utils32_pkg" >/dev/null 2>&1; then yes | sudo pacman -Sy --noconfirm "$utils32_pkg" || true; fi
|
|
fi
|
|
}
|
|
|
|
echo "Detected NVIDIA GPU. Selecting driver..."
|
|
NVIDIA_DRIVER_PACKAGE=$(_choose_nvidia_pkg)
|
|
export NVIDIA_DRIVER_PACKAGE
|
|
_remove_conflicting_nvidia_pkgs "$NVIDIA_DRIVER_PACKAGE"
|
|
_install_nvidia_stack "$NVIDIA_DRIVER_PACKAGE"
|
|
export SKIP_NVIDIA_PACKAGES="false"
|
|
echo "NVIDIA driver installation finished (package: $NVIDIA_DRIVER_PACKAGE)"
|
|
echo "Optional: adjust /etc/mkinitcpio.conf (remove kms) then: sudo mkinitcpio -P"
|