mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2026-04-25 08:56:21 +00:00
update nvidia_installer.sh
This commit is contained in:
MacRimi
@@ -20,6 +20,12 @@ screen_capture="/tmp/proxmenux_nvidia_screen_capture_$$.txt"
|
|||||||
NVIDIA_BASE_URL="https://download.nvidia.com/XFree86/Linux-x86_64"
|
NVIDIA_BASE_URL="https://download.nvidia.com/XFree86/Linux-x86_64"
|
||||||
NVIDIA_WORKDIR="/opt/nvidia"
|
NVIDIA_WORKDIR="/opt/nvidia"
|
||||||
|
|
||||||
|
# LXC post-install update constants (used only when NVIDIA LXC passthrough
|
||||||
|
# containers are detected and the user confirms updating them after the host
|
||||||
|
# install/reinstall finishes).
|
||||||
|
NVIDIA_INSTALL_MIN_MB=2048
|
||||||
|
CT_ORIG_MEM=""
|
||||||
|
|
||||||
export BASE_DIR
|
export BASE_DIR
|
||||||
export COMPONENTS_STATUS_FILE
|
export COMPONENTS_STATUS_FILE
|
||||||
|
|
||||||
@@ -119,6 +125,272 @@ detect_driver_status() {
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# ==========================================================
|
||||||
|
# LXC NVIDIA passthrough — discovery & userspace-libs update
|
||||||
|
# Invoked after the host install/reinstall completes. Aligned with the install
|
||||||
|
# path used in add_gpu_lxc.sh (distro-aware, memory/disk checks, --no-dkms,
|
||||||
|
# --no-install-compat32-libs, visible progress via tee).
|
||||||
|
# ==========================================================
|
||||||
|
find_nvidia_containers() {
|
||||||
|
NVIDIA_CONTAINERS=()
|
||||||
|
for conf in /etc/pve/lxc/*.conf; do
|
||||||
|
[[ -f "$conf" ]] || continue
|
||||||
|
if grep -qiE "dev[0-9]+:.*nvidia" "$conf"; then
|
||||||
|
NVIDIA_CONTAINERS+=("$(basename "$conf" .conf)")
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
get_lxc_nvidia_version() {
|
||||||
|
local ctid="$1"
|
||||||
|
local version=""
|
||||||
|
|
||||||
|
# Prefer nvidia-smi when the container is running (works with .run-installed drivers)
|
||||||
|
if pct status "$ctid" 2>/dev/null | grep -q "running"; then
|
||||||
|
version=$(pct exec "$ctid" -- nvidia-smi \
|
||||||
|
--query-gpu=driver_version --format=csv,noheader 2>/dev/null \
|
||||||
|
| head -1 | tr -d '[:space:]' || true)
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Fallback: dpkg status for apt-installed libcuda1 (dir-type storage, no start needed)
|
||||||
|
if [[ -z "$version" ]]; then
|
||||||
|
local rootfs="/var/lib/lxc/${ctid}/rootfs"
|
||||||
|
if [[ -f "${rootfs}/var/lib/dpkg/status" ]]; then
|
||||||
|
version=$(grep -A5 "^Package: libcuda1$" "${rootfs}/var/lib/dpkg/status" \
|
||||||
|
| grep "^Version:" | head -1 | awk '{print $2}' | cut -d- -f1)
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "${version:-$(translate 'not installed')}"
|
||||||
|
}
|
||||||
|
|
||||||
|
_detect_container_distro() {
|
||||||
|
local distro
|
||||||
|
distro=$(pct exec "$1" -- grep "^ID=" /etc/os-release 2>/dev/null \
|
||||||
|
| cut -d= -f2 | tr -d '[:space:]"')
|
||||||
|
echo "${distro:-unknown}"
|
||||||
|
}
|
||||||
|
|
||||||
|
_ensure_container_memory() {
|
||||||
|
local ctid="$1"
|
||||||
|
local cur_mem
|
||||||
|
cur_mem=$(pct config "$ctid" 2>/dev/null | awk '/^memory:/{print $2}')
|
||||||
|
[[ -z "$cur_mem" ]] && cur_mem=512
|
||||||
|
|
||||||
|
if [[ "$cur_mem" -lt "$NVIDIA_INSTALL_MIN_MB" ]]; then
|
||||||
|
if whiptail --title "$(translate 'Low Container Memory')" --yesno \
|
||||||
|
"$(translate 'Container') ${ctid} $(translate 'has') ${cur_mem}MB RAM.\n\n$(translate 'The NVIDIA installer needs at least') ${NVIDIA_INSTALL_MIN_MB}MB $(translate 'to run without being killed by the OOM killer.')\n\n$(translate 'Increase container RAM temporarily to') ${NVIDIA_INSTALL_MIN_MB}MB?" \
|
||||||
|
13 72; then
|
||||||
|
CT_ORIG_MEM="$cur_mem"
|
||||||
|
pct set "$ctid" -memory "$NVIDIA_INSTALL_MIN_MB" >>"$LOG_FILE" 2>&1 || true
|
||||||
|
else
|
||||||
|
msg_warn "$(translate 'Insufficient memory. Skipping LXC') ${ctid}."
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
_restore_container_memory() {
|
||||||
|
local ctid="$1"
|
||||||
|
if [[ -n "$CT_ORIG_MEM" ]]; then
|
||||||
|
msg_info "$(translate 'Restoring container memory to') ${CT_ORIG_MEM}MB..."
|
||||||
|
pct set "$ctid" -memory "$CT_ORIG_MEM" >>"$LOG_FILE" 2>&1 || true
|
||||||
|
msg_ok "$(translate 'Memory restored.')"
|
||||||
|
CT_ORIG_MEM=""
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
_start_container_and_wait() {
|
||||||
|
local ctid="$1"
|
||||||
|
msg_info "$(translate 'Starting container') ${ctid}..."
|
||||||
|
pct start "$ctid" >>"$LOG_FILE" 2>&1 || true
|
||||||
|
|
||||||
|
local ready=false
|
||||||
|
for _ in {1..15}; do
|
||||||
|
sleep 2
|
||||||
|
if pct exec "$ctid" -- true >/dev/null 2>&1; then
|
||||||
|
ready=true
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
if ! $ready; then
|
||||||
|
msg_warn "$(translate 'Container') ${ctid} $(translate 'did not become ready. Skipping.')"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
msg_ok "$(translate 'Container') ${ctid} $(translate 'started.')" | tee -a "$screen_capture"
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
update_lxc_nvidia() {
|
||||||
|
local ctid="$1"
|
||||||
|
local version="$2"
|
||||||
|
local started_here=false
|
||||||
|
|
||||||
|
local old_version
|
||||||
|
old_version=$(get_lxc_nvidia_version "$ctid")
|
||||||
|
|
||||||
|
msg_info2 "$(translate 'Container') ${ctid}: $(translate 'updating NVIDIA userspace libs') (${old_version} → ${version})"
|
||||||
|
|
||||||
|
if ! pct status "$ctid" 2>/dev/null | grep -q "running"; then
|
||||||
|
started_here=true
|
||||||
|
_start_container_and_wait "$ctid" || return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
msg_info "$(translate 'Detecting container OS...')"
|
||||||
|
local distro
|
||||||
|
distro=$(_detect_container_distro "$ctid")
|
||||||
|
msg_ok "$(translate 'Container OS:') ${distro}" | tee -a "$screen_capture"
|
||||||
|
|
||||||
|
local install_rc=0
|
||||||
|
|
||||||
|
case "$distro" in
|
||||||
|
alpine)
|
||||||
|
msg_info2 "$(translate 'Upgrading NVIDIA utils (Alpine)...')"
|
||||||
|
pct exec "$ctid" -- sh -c \
|
||||||
|
"apk update && apk add --no-cache --upgrade nvidia-utils" \
|
||||||
|
2>&1 | tee -a "$LOG_FILE"
|
||||||
|
install_rc=${PIPESTATUS[0]}
|
||||||
|
;;
|
||||||
|
arch|manjaro|endeavouros)
|
||||||
|
msg_info2 "$(translate 'Upgrading NVIDIA utils (Arch)...')"
|
||||||
|
pct exec "$ctid" -- bash -c \
|
||||||
|
"pacman -Syu --noconfirm nvidia-utils" \
|
||||||
|
2>&1 | tee -a "$LOG_FILE"
|
||||||
|
install_rc=${PIPESTATUS[0]}
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
local run_file="${NVIDIA_WORKDIR}/NVIDIA-Linux-x86_64-${version}.run"
|
||||||
|
|
||||||
|
if [[ ! -f "$run_file" ]]; then
|
||||||
|
msg_warn "$(translate 'Installer not found:') ${run_file}. $(translate 'Skipping LXC') ${ctid}."
|
||||||
|
install_rc=1
|
||||||
|
elif ! _ensure_container_memory "$ctid"; then
|
||||||
|
install_rc=1
|
||||||
|
else
|
||||||
|
local free_mb
|
||||||
|
free_mb=$(pct exec "$ctid" -- df -m / 2>/dev/null | awk 'NR==2{print $4}' || echo 0)
|
||||||
|
if [[ "$free_mb" -lt 1500 ]]; then
|
||||||
|
_restore_container_memory "$ctid"
|
||||||
|
dialog --backtitle "ProxMenux" \
|
||||||
|
--title "$(translate 'Insufficient Disk Space')" \
|
||||||
|
--msgbox "\n$(translate 'Container') ${ctid} $(translate 'has only') ${free_mb}MB $(translate 'of free disk space.')\n\n$(translate 'NVIDIA libs require approximately 1.5GB of free space.')" \
|
||||||
|
11 72
|
||||||
|
msg_warn "$(translate 'Insufficient disk space. Skipping LXC') ${ctid}."
|
||||||
|
install_rc=1
|
||||||
|
else
|
||||||
|
local extract_dir="${NVIDIA_WORKDIR}/extracted_${version}"
|
||||||
|
local archive="/tmp/nvidia_lxc_${version}.tar.gz"
|
||||||
|
|
||||||
|
msg_info2 "$(translate 'Extracting NVIDIA installer on host...')"
|
||||||
|
rm -rf "$extract_dir"
|
||||||
|
sh "$run_file" --extract-only --target "$extract_dir" 2>&1 | tee -a "$LOG_FILE"
|
||||||
|
if [[ ${PIPESTATUS[0]} -ne 0 ]]; then
|
||||||
|
msg_warn "$(translate 'Extraction failed. Check log:') ${LOG_FILE}"
|
||||||
|
_restore_container_memory "$ctid"
|
||||||
|
install_rc=1
|
||||||
|
else
|
||||||
|
msg_ok "$(translate 'NVIDIA installer extracted.')" | tee -a "$screen_capture"
|
||||||
|
|
||||||
|
msg_info2 "$(translate 'Packing installer archive...')"
|
||||||
|
tar --checkpoint=5000 --checkpoint-action=dot \
|
||||||
|
-czf "$archive" -C "$extract_dir" . 2>&1 | tee -a "$LOG_FILE"
|
||||||
|
echo ""
|
||||||
|
local archive_size
|
||||||
|
archive_size=$(du -sh "$archive" 2>/dev/null | cut -f1)
|
||||||
|
msg_ok "$(translate 'Archive ready') (${archive_size})." | tee -a "$screen_capture"
|
||||||
|
|
||||||
|
msg_info "$(translate 'Copying installer to container') ${ctid}..."
|
||||||
|
if ! pct push "$ctid" "$archive" /tmp/nvidia_lxc.tar.gz >>"$LOG_FILE" 2>&1; then
|
||||||
|
msg_warn "$(translate 'pct push failed. Check log:') ${LOG_FILE}"
|
||||||
|
rm -f "$archive"
|
||||||
|
rm -rf "$extract_dir"
|
||||||
|
_restore_container_memory "$ctid"
|
||||||
|
install_rc=1
|
||||||
|
else
|
||||||
|
rm -f "$archive"
|
||||||
|
msg_ok "$(translate 'Installer copied to container.')" | tee -a "$screen_capture"
|
||||||
|
|
||||||
|
msg_info2 "$(translate 'Running NVIDIA installer in container. This may take several minutes...')"
|
||||||
|
echo "" >>"$LOG_FILE"
|
||||||
|
pct exec "$ctid" -- bash -c "
|
||||||
|
mkdir -p /tmp/nvidia_lxc_install
|
||||||
|
tar -xzf /tmp/nvidia_lxc.tar.gz -C /tmp/nvidia_lxc_install 2>&1
|
||||||
|
/tmp/nvidia_lxc_install/nvidia-installer \
|
||||||
|
--no-kernel-modules \
|
||||||
|
--no-questions \
|
||||||
|
--ui=none \
|
||||||
|
--no-nouveau-check \
|
||||||
|
--no-dkms \
|
||||||
|
--no-install-compat32-libs
|
||||||
|
EXIT=\$?
|
||||||
|
rm -rf /tmp/nvidia_lxc_install /tmp/nvidia_lxc.tar.gz
|
||||||
|
exit \$EXIT
|
||||||
|
" 2>&1 | tee -a "$LOG_FILE"
|
||||||
|
install_rc=${PIPESTATUS[0]}
|
||||||
|
|
||||||
|
rm -rf "$extract_dir"
|
||||||
|
_restore_container_memory "$ctid"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
if [[ $install_rc -ne 0 ]]; then
|
||||||
|
msg_warn "$(translate 'NVIDIA update failed for LXC') ${ctid} (rc=${install_rc}). $(translate 'Check log:') ${LOG_FILE}"
|
||||||
|
if $started_here; then
|
||||||
|
pct stop "$ctid" >>"$LOG_FILE" 2>&1 || true
|
||||||
|
fi
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if pct exec "$ctid" -- sh -c "which nvidia-smi" >/dev/null 2>&1; then
|
||||||
|
local new_ver
|
||||||
|
new_ver=$(pct exec "$ctid" -- nvidia-smi \
|
||||||
|
--query-gpu=driver_version --format=csv,noheader 2>/dev/null \
|
||||||
|
| head -1 | tr -d '[:space:]' || true)
|
||||||
|
msg_ok "$(translate 'Container') ${ctid}: ${old_version} → ${new_ver:-$version}" | tee -a "$screen_capture"
|
||||||
|
else
|
||||||
|
msg_warn "$(translate 'nvidia-smi not found in container') ${ctid} $(translate 'after update.')"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if $started_here; then
|
||||||
|
msg_info "$(translate 'Stopping container') ${ctid}..."
|
||||||
|
pct stop "$ctid" >>"$LOG_FILE" 2>&1 || true
|
||||||
|
msg_ok "$(translate 'Container stopped.')" | tee -a "$screen_capture"
|
||||||
|
fi
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
# Post-host-install LXC update offer — scans for NVIDIA LXCs and, if any are
|
||||||
|
# found, asks the user if they want to propagate the driver update to them.
|
||||||
|
offer_lxc_updates_if_any() {
|
||||||
|
local target_version="$1"
|
||||||
|
find_nvidia_containers
|
||||||
|
[[ ${#NVIDIA_CONTAINERS[@]} -eq 0 ]] && return 0
|
||||||
|
|
||||||
|
local info ctid lxc_ver ct_name
|
||||||
|
info="\n$(translate 'The following LXC containers have NVIDIA passthrough configured:')\n\n"
|
||||||
|
for ctid in "${NVIDIA_CONTAINERS[@]}"; do
|
||||||
|
lxc_ver=$(get_lxc_nvidia_version "$ctid")
|
||||||
|
ct_name=$(pct config "$ctid" 2>/dev/null | grep "^hostname:" | awk '{print $2}')
|
||||||
|
info+=" CT ${ctid} ${ct_name:+(${ct_name})} — $(translate 'driver:') ${lxc_ver}\n"
|
||||||
|
done
|
||||||
|
info+="\n$(translate 'Do you want to update the NVIDIA userspace libraries inside these containers to match the host?')"
|
||||||
|
|
||||||
|
if ! hybrid_yesno "$(translate 'Update NVIDIA in LXC Containers')" "$info" 20 80; then
|
||||||
|
msg_info2 "$(translate 'LXC update skipped by user.')"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
for ctid in "${NVIDIA_CONTAINERS[@]}"; do
|
||||||
|
update_lxc_nvidia "$ctid" "$target_version" || true
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
# ==========================================================
|
# ==========================================================
|
||||||
# System preparation (repos, headers, etc.)
|
# System preparation (repos, headers, etc.)
|
||||||
# ==========================================================
|
# ==========================================================
|
||||||
@@ -533,9 +805,21 @@ download_nvidia_installer() {
|
|||||||
"${NVIDIA_BASE_URL}/${version}/NVIDIA-Linux-x86_64-${version}-no-compat32.run"
|
"${NVIDIA_BASE_URL}/${version}/NVIDIA-Linux-x86_64-${version}-no-compat32.run"
|
||||||
)
|
)
|
||||||
|
|
||||||
# Header line on the real terminal so it stays visible regardless of caller redirects.
|
# Web mode (ProxMenux Monitor) runs scripts without a controlling TTY, so
|
||||||
printf '\n %s NVIDIA-Linux-x86_64-%s.run\n' \
|
# /dev/tty is not writable and progress-bar animations using \r don't render
|
||||||
"$(translate 'Downloading')" "$version" >/dev/tty
|
# in the web terminal. Fall back to a quiet wget in that case; interactive
|
||||||
|
# users (SSH / console) still get the ISO-like progress bar.
|
||||||
|
local _nv_has_tty=false
|
||||||
|
if ! is_web_mode 2>/dev/null && [[ -t 2 ]]; then
|
||||||
|
_nv_has_tty=true
|
||||||
|
fi
|
||||||
|
|
||||||
|
if $_nv_has_tty; then
|
||||||
|
printf '\n %s NVIDIA-Linux-x86_64-%s.run\n' \
|
||||||
|
"$(translate 'Downloading')" "$version" >/dev/tty
|
||||||
|
else
|
||||||
|
echo " $(translate 'Downloading') NVIDIA-Linux-x86_64-${version}.run" >&2
|
||||||
|
fi
|
||||||
|
|
||||||
local success=false
|
local success=false
|
||||||
local url_index=0
|
local url_index=0
|
||||||
@@ -546,12 +830,24 @@ download_nvidia_installer() {
|
|||||||
|
|
||||||
rm -f "$run_file"
|
rm -f "$run_file"
|
||||||
|
|
||||||
# wget --show-progress writes its progress bar to stderr. We route it to
|
local _dl_ok=false
|
||||||
# /dev/tty explicitly so the user always sees it (same UX as ISO downloads
|
if $_nv_has_tty; then
|
||||||
# in vm_creator.sh). The file contents still go to $run_file.
|
# Interactive: progress bar to /dev/tty (bypasses any caller redirection).
|
||||||
if wget --no-verbose --show-progress \
|
if wget --no-verbose --show-progress \
|
||||||
--connect-timeout=30 --timeout=600 --tries=1 \
|
--connect-timeout=30 --timeout=600 --tries=1 \
|
||||||
-O "$run_file" "$url" 2>/dev/tty; then
|
-O "$run_file" "$url" 2>/dev/tty; then
|
||||||
|
_dl_ok=true
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
# Web / no-TTY: silent wget, log errors only.
|
||||||
|
if wget --quiet \
|
||||||
|
--connect-timeout=30 --timeout=600 --tries=1 \
|
||||||
|
-O "$run_file" "$url" 2>>"$LOG_FILE"; then
|
||||||
|
_dl_ok=true
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if $_dl_ok; then
|
||||||
echo "Download completed, verifying file..." >> "$LOG_FILE"
|
echo "Download completed, verifying file..." >> "$LOG_FILE"
|
||||||
|
|
||||||
if [[ ! -f "$run_file" ]]; then
|
if [[ ! -f "$run_file" ]]; then
|
||||||
@@ -752,18 +1048,32 @@ show_install_overview() {
|
|||||||
overview+=" • $(translate 'Install NVIDIA proprietary drivers')\n"
|
overview+=" • $(translate 'Install NVIDIA proprietary drivers')\n"
|
||||||
overview+=" • $(translate 'Configure GPU passthrough with VFIO')\n"
|
overview+=" • $(translate 'Configure GPU passthrough with VFIO')\n"
|
||||||
overview+=" • $(translate 'Blacklist nouveau driver')\n"
|
overview+=" • $(translate 'Blacklist nouveau driver')\n"
|
||||||
overview+=" • $(translate 'Enable IOMMU support if not enabled')\n\n"
|
overview+=" • $(translate 'Enable IOMMU support if not enabled')\n"
|
||||||
|
overview+=" • $(translate 'Optionally update NVIDIA libs in LXC containers with passthrough')\n\n"
|
||||||
|
|
||||||
overview+="$(translate 'Detected GPU(s):')\n"
|
overview+="$(translate 'Detected GPU(s):')\n"
|
||||||
overview+="\Zb\Z4$DETECTED_GPUS_TEXT\Zn\n"
|
overview+="\Zb\Z4$DETECTED_GPUS_TEXT\Zn\n"
|
||||||
|
|
||||||
overview+="\n\Zn$(translate 'Current status: ') "
|
overview+="\n\Zn$(translate 'Current status: ') "
|
||||||
overview+="\Zb${CURRENT_STATUS_TEXT}\Zn\n\n"
|
overview+="\Zb${CURRENT_STATUS_TEXT}\Zn\n"
|
||||||
|
|
||||||
overview+="$(translate 'After confirming, you will be asked to choose the NVIDIA driver version to install.')\n\n"
|
# Scan for LXC containers with NVIDIA passthrough and surface them in the
|
||||||
|
# overview so the user knows upfront they will be offered a driver update.
|
||||||
|
find_nvidia_containers
|
||||||
|
if [[ ${#NVIDIA_CONTAINERS[@]} -gt 0 ]]; then
|
||||||
|
overview+="\n$(translate 'LXC containers with NVIDIA passthrough:')\n"
|
||||||
|
local ctid lxc_ver ct_name
|
||||||
|
for ctid in "${NVIDIA_CONTAINERS[@]}"; do
|
||||||
|
lxc_ver=$(get_lxc_nvidia_version "$ctid")
|
||||||
|
ct_name=$(pct config "$ctid" 2>/dev/null | grep "^hostname:" | awk '{print $2}')
|
||||||
|
overview+=" \Zb\Z4CT ${ctid}\Zn ${ct_name:+(${ct_name})} — $(translate 'driver:') ${lxc_ver}\n"
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
overview+="\n$(translate 'After confirming, you will be asked to choose the NVIDIA driver version to install.')\n\n"
|
||||||
overview+="$(translate 'Do you want to continue?')"
|
overview+="$(translate 'Do you want to continue?')"
|
||||||
|
|
||||||
hybrid_yesno "$(translate 'NVIDIA GPU Driver Installation')" "$overview" 22 90
|
hybrid_yesno "$(translate 'NVIDIA GPU Driver Installation')" "$overview" 24 90
|
||||||
}
|
}
|
||||||
|
|
||||||
show_version_menu() {
|
show_version_menu() {
|
||||||
@@ -991,6 +1301,13 @@ main() {
|
|||||||
update_component_status "nvidia_driver" "failed" "" "gpu" '{"patched":false}'
|
update_component_status "nvidia_driver" "failed" "" "gpu" '{"patched":false}'
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# Propagate the new driver to LXC containers with NVIDIA passthrough, if any.
|
||||||
|
# Uses the same .run installer cached in $NVIDIA_WORKDIR — runs only if the
|
||||||
|
# host install succeeded and the user confirms.
|
||||||
|
if [[ -n "$CURRENT_DRIVER_VERSION" ]]; then
|
||||||
|
offer_lxc_updates_if_any "$CURRENT_DRIVER_VERSION"
|
||||||
|
fi
|
||||||
|
|
||||||
apply_nvidia_patch_if_needed
|
apply_nvidia_patch_if_needed
|
||||||
restart_prompt
|
restart_prompt
|
||||||
;;
|
;;
|
||||||
|
|||||||
@@ -1,947 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
# ProxMenux - NVIDIA Driver Updater (Host + LXC)
|
|
||||||
# ================================================
|
|
||||||
# Author : MacRimi
|
|
||||||
# License : MIT
|
|
||||||
# Version : 2.0
|
|
||||||
# Last Updated: 17/04/2026
|
|
||||||
# ================================================
|
|
||||||
#
|
|
||||||
# Aligned with nvidia_installer.sh (host install flow & kernel filter)
|
|
||||||
# and add_gpu_lxc.sh (LXC userspace install flow with distro + memory
|
|
||||||
# awareness and visible progress output).
|
|
||||||
|
|
||||||
SCRIPT_TITLE="NVIDIA Driver Update (Host + LXC)"
|
|
||||||
|
|
||||||
LOCAL_SCRIPTS="/usr/local/share/proxmenux/scripts"
|
|
||||||
BASE_DIR="/usr/local/share/proxmenux"
|
|
||||||
UTILS_FILE="$BASE_DIR/utils.sh"
|
|
||||||
COMPONENTS_STATUS_FILE="$BASE_DIR/components_status.json"
|
|
||||||
LOG_FILE="/tmp/nvidia_update.log"
|
|
||||||
screen_capture="/tmp/proxmenux_nvidia_update_screen_capture_$$.txt"
|
|
||||||
|
|
||||||
NVIDIA_BASE_URL="https://download.nvidia.com/XFree86/Linux-x86_64"
|
|
||||||
NVIDIA_WORKDIR="/opt/nvidia"
|
|
||||||
|
|
||||||
export BASE_DIR
|
|
||||||
export COMPONENTS_STATUS_FILE
|
|
||||||
|
|
||||||
if [[ -f "$UTILS_FILE" ]]; then
|
|
||||||
source "$UTILS_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ ! -f "$COMPONENTS_STATUS_FILE" ]]; then
|
|
||||||
echo "{}" > "$COMPONENTS_STATUS_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
load_language
|
|
||||||
initialize_cache
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# GPU passthrough guard — block update when GPU is in VM passthrough mode
|
|
||||||
# ============================================================
|
|
||||||
check_gpu_not_in_vm_passthrough() {
|
|
||||||
local dev vendor driver vfio_list=""
|
|
||||||
for dev in /sys/bus/pci/devices/*; do
|
|
||||||
vendor=$(cat "$dev/vendor" 2>/dev/null)
|
|
||||||
[[ "$vendor" != "0x10de" ]] && continue
|
|
||||||
if [[ -L "$dev/driver" ]]; then
|
|
||||||
driver=$(basename "$(readlink "$dev/driver")")
|
|
||||||
if [[ "$driver" == "vfio-pci" ]]; then
|
|
||||||
vfio_list+=" • $(basename "$dev")\n"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
[[ -z "$vfio_list" ]] && return 0
|
|
||||||
|
|
||||||
local msg
|
|
||||||
msg="\n$(translate "One or more NVIDIA GPUs are currently configured for VM passthrough (vfio-pci):")\n\n"
|
|
||||||
msg+="${vfio_list}\n"
|
|
||||||
msg+="$(translate "Updating host drivers while the GPU is assigned to a VM could break passthrough and destabilize the system.")\n\n"
|
|
||||||
msg+="$(translate "To update host drivers, first remove the GPU from VM passthrough configuration and reboot.")"
|
|
||||||
|
|
||||||
dialog --backtitle "ProxMenux" \
|
|
||||||
--title "$(translate "GPU in VM Passthrough Mode")" \
|
|
||||||
--msgbox "$msg" 16 78
|
|
||||||
exit 0
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Host NVIDIA state detection
|
|
||||||
# ============================================================
|
|
||||||
detect_host_nvidia() {
|
|
||||||
HOST_NVIDIA_VERSION=""
|
|
||||||
HOST_NVIDIA_READY=false
|
|
||||||
|
|
||||||
if lsmod | grep -q "^nvidia " && command -v nvidia-smi >/dev/null 2>&1; then
|
|
||||||
HOST_NVIDIA_VERSION=$(nvidia-smi --query-gpu=driver_version \
|
|
||||||
--format=csv,noheader 2>/dev/null | head -n1 | tr -d '[:space:]')
|
|
||||||
[[ -n "$HOST_NVIDIA_VERSION" ]] && HOST_NVIDIA_READY=true
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! $HOST_NVIDIA_READY; then
|
|
||||||
dialog --backtitle "ProxMenux" \
|
|
||||||
--title "$(translate 'NVIDIA Not Found')" \
|
|
||||||
--msgbox "\n$(translate 'No NVIDIA driver is currently loaded on this host.')\n\n$(translate 'Please install NVIDIA drivers first using the option:')\n\n $(translate 'Install NVIDIA Drivers on Host')\n\n$(translate 'from this same GPU and TPU menu.')" \
|
|
||||||
13 72
|
|
||||||
exit 0
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Kernel compatibility (mirror of nvidia_installer.sh)
|
|
||||||
# ============================================================
|
|
||||||
get_kernel_compatibility_info() {
|
|
||||||
local kernel_version
|
|
||||||
kernel_version=$(uname -r)
|
|
||||||
|
|
||||||
KERNEL_MAJOR=$(echo "$kernel_version" | cut -d. -f1)
|
|
||||||
KERNEL_MINOR=$(echo "$kernel_version" | cut -d. -f2)
|
|
||||||
|
|
||||||
# Minimum driver version matrix (keyed to kernel series) — based on
|
|
||||||
# https://docs.nvidia.com/datacenter/tesla/drivers/index.html
|
|
||||||
if [[ "$KERNEL_MAJOR" -ge 6 ]] && [[ "$KERNEL_MINOR" -ge 17 ]]; then
|
|
||||||
MIN_DRIVER_VERSION="580.82.07" # PVE 9.x
|
|
||||||
elif [[ "$KERNEL_MAJOR" -ge 6 ]] && [[ "$KERNEL_MINOR" -ge 8 ]]; then
|
|
||||||
MIN_DRIVER_VERSION="550" # PVE 8.2+
|
|
||||||
elif [[ "$KERNEL_MAJOR" -ge 6 ]]; then
|
|
||||||
MIN_DRIVER_VERSION="535" # PVE 8.x initial
|
|
||||||
elif [[ "$KERNEL_MAJOR" -eq 5 ]] && [[ "$KERNEL_MINOR" -ge 15 ]]; then
|
|
||||||
MIN_DRIVER_VERSION="470" # PVE 7.x / 8.x legacy
|
|
||||||
else
|
|
||||||
MIN_DRIVER_VERSION="450" # Old kernels
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
is_version_compatible() {
|
|
||||||
local version="$1"
|
|
||||||
local ver_major ver_minor ver_patch
|
|
||||||
|
|
||||||
ver_major=$(echo "$version" | cut -d. -f1)
|
|
||||||
ver_minor=$(echo "$version" | cut -d. -f2)
|
|
||||||
ver_patch=$(echo "$version" | cut -d. -f3)
|
|
||||||
|
|
||||||
if [[ "$MIN_DRIVER_VERSION" == "580.82.07" ]]; then
|
|
||||||
if [[ ${ver_major} -gt 580 ]]; then
|
|
||||||
return 0
|
|
||||||
elif [[ ${ver_major} -eq 580 ]]; then
|
|
||||||
if [[ $((10#${ver_minor})) -gt 82 ]]; then
|
|
||||||
return 0
|
|
||||||
elif [[ $((10#${ver_minor})) -eq 82 ]]; then
|
|
||||||
if [[ $((10#${ver_patch:-0})) -ge 7 ]]; then
|
|
||||||
return 0
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ ${ver_major} -ge ${MIN_DRIVER_VERSION} ]]; then
|
|
||||||
return 0
|
|
||||||
else
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
version_le() {
|
|
||||||
local v1="$1"
|
|
||||||
local v2="$2"
|
|
||||||
|
|
||||||
IFS='.' read -r a1 b1 c1 <<<"$v1"
|
|
||||||
IFS='.' read -r a2 b2 c2 <<<"$v2"
|
|
||||||
|
|
||||||
a1=${a1:-0}; b1=${b1:-0}; c1=${c1:-0}
|
|
||||||
a2=${a2:-0}; b2=${b2:-0}; c2=${c2:-0}
|
|
||||||
|
|
||||||
a1=$((10#$a1)); b1=$((10#$b1)); c1=$((10#$c1))
|
|
||||||
a2=$((10#$a2)); b2=$((10#$b2)); c2=$((10#$c2))
|
|
||||||
|
|
||||||
if (( a1 < a2 )); then
|
|
||||||
return 0
|
|
||||||
elif (( a1 > a2 )); then
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
if (( b1 < b2 )); then
|
|
||||||
return 0
|
|
||||||
elif (( b1 > b2 )); then
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
if (( c1 <= c2 )); then
|
|
||||||
return 0
|
|
||||||
else
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# LXC containers with NVIDIA passthrough
|
|
||||||
# ============================================================
|
|
||||||
find_nvidia_containers() {
|
|
||||||
NVIDIA_CONTAINERS=()
|
|
||||||
for conf in /etc/pve/lxc/*.conf; do
|
|
||||||
[[ -f "$conf" ]] || continue
|
|
||||||
if grep -qiE "dev[0-9]+:.*nvidia" "$conf"; then
|
|
||||||
NVIDIA_CONTAINERS+=("$(basename "$conf" .conf)")
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
}
|
|
||||||
|
|
||||||
get_lxc_nvidia_version() {
|
|
||||||
local ctid="$1"
|
|
||||||
local version=""
|
|
||||||
|
|
||||||
# Prefer nvidia-smi when the container is running (works with .run-installed drivers)
|
|
||||||
if pct status "$ctid" 2>/dev/null | grep -q "running"; then
|
|
||||||
version=$(pct exec "$ctid" -- nvidia-smi \
|
|
||||||
--query-gpu=driver_version --format=csv,noheader 2>/dev/null \
|
|
||||||
| head -1 | tr -d '[:space:]' || true)
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Fallback: dpkg status for apt-installed libcuda1 (dir-type storage, no start needed)
|
|
||||||
if [[ -z "$version" ]]; then
|
|
||||||
local rootfs="/var/lib/lxc/${ctid}/rootfs"
|
|
||||||
if [[ -f "${rootfs}/var/lib/dpkg/status" ]]; then
|
|
||||||
version=$(grep -A5 "^Package: libcuda1$" "${rootfs}/var/lib/dpkg/status" \
|
|
||||||
| grep "^Version:" | head -1 | awk '{print $2}' | cut -d- -f1)
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo "${version:-$(translate 'not installed')}"
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Version list from NVIDIA servers
|
|
||||||
# ============================================================
|
|
||||||
list_available_versions() {
|
|
||||||
local html
|
|
||||||
html=$(curl -s --connect-timeout 15 "${NVIDIA_BASE_URL}/" 2>/dev/null) || true
|
|
||||||
|
|
||||||
if [[ -z "$html" ]]; then
|
|
||||||
echo ""
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo "$html" \
|
|
||||||
| grep -o 'href=[^ >]*' \
|
|
||||||
| awk -F"'" '{print $2}' \
|
|
||||||
| grep -E '^[0-9]' \
|
|
||||||
| sed 's/\/$//' \
|
|
||||||
| sed "s/^[[:space:]]*//;s/[[:space:]]*$//" \
|
|
||||||
| sort -Vr \
|
|
||||||
| uniq
|
|
||||||
}
|
|
||||||
|
|
||||||
get_latest_version() {
|
|
||||||
local latest_line
|
|
||||||
latest_line=$(curl -fsSL --connect-timeout 15 "${NVIDIA_BASE_URL}/latest.txt" 2>/dev/null) || true
|
|
||||||
echo "$latest_line" | awk '{print $1}' | tr -d '[:space:]'
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Version selection menu (filtered by kernel compatibility)
|
|
||||||
# ============================================================
|
|
||||||
select_target_version() {
|
|
||||||
local latest versions_list
|
|
||||||
latest=$(get_latest_version 2>/dev/null)
|
|
||||||
versions_list=$(list_available_versions 2>/dev/null)
|
|
||||||
|
|
||||||
if [[ -z "$latest" && -z "$versions_list" ]]; then
|
|
||||||
dialog --backtitle "ProxMenux" \
|
|
||||||
--title "$(translate 'Error')" \
|
|
||||||
--msgbox "\n$(translate 'Could not retrieve versions from NVIDIA. Please check your internet connection.')" \
|
|
||||||
8 72
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
[[ -z "$latest" && -n "$versions_list" ]] && latest=$(echo "$versions_list" | head -1)
|
|
||||||
[[ -z "$versions_list" ]] && versions_list="$latest"
|
|
||||||
latest=$(echo "$latest" | tr -d '[:space:]')
|
|
||||||
|
|
||||||
# Kernel compatibility filter
|
|
||||||
local current_list="$versions_list"
|
|
||||||
if [[ -n "$MIN_DRIVER_VERSION" ]]; then
|
|
||||||
local filtered_list=""
|
|
||||||
while IFS= read -r ver; do
|
|
||||||
[[ -z "$ver" ]] && continue
|
|
||||||
if is_version_compatible "$ver"; then
|
|
||||||
filtered_list+="$ver"$'\n'
|
|
||||||
fi
|
|
||||||
done <<< "$current_list"
|
|
||||||
current_list="$filtered_list"
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Cap at latest (avoid showing beta branches newer than published latest)
|
|
||||||
if [[ -n "$latest" ]]; then
|
|
||||||
local filtered_max_list=""
|
|
||||||
while IFS= read -r ver; do
|
|
||||||
[[ -z "$ver" ]] && continue
|
|
||||||
if version_le "$ver" "$latest"; then
|
|
||||||
filtered_max_list+="$ver"$'\n'
|
|
||||||
fi
|
|
||||||
done <<< "$current_list"
|
|
||||||
current_list="$filtered_max_list"
|
|
||||||
fi
|
|
||||||
|
|
||||||
local menu_text
|
|
||||||
menu_text="\n$(translate 'Current host version:') ${HOST_NVIDIA_VERSION}\n"
|
|
||||||
menu_text+="$(translate 'Kernel:') $(uname -r)\n\n"
|
|
||||||
menu_text+="$(translate 'Select the target version to install on host and all affected LXCs:')\n"
|
|
||||||
menu_text+="$(translate 'Versions shown are compatible with your running kernel.')"
|
|
||||||
|
|
||||||
local choices=()
|
|
||||||
choices+=("latest" "$(translate 'Latest available') (${latest:-?})")
|
|
||||||
choices+=("" "")
|
|
||||||
|
|
||||||
if [[ -n "$current_list" ]]; then
|
|
||||||
while IFS= read -r ver; do
|
|
||||||
ver=$(echo "$ver" | tr -d '[:space:]')
|
|
||||||
[[ -z "$ver" ]] && continue
|
|
||||||
choices+=("$ver" "$ver")
|
|
||||||
done <<< "$current_list"
|
|
||||||
else
|
|
||||||
choices+=("" "$(translate 'No compatible versions found for your kernel')")
|
|
||||||
fi
|
|
||||||
|
|
||||||
TARGET_VERSION=$(dialog --backtitle "ProxMenux" \
|
|
||||||
--title "$(translate 'NVIDIA Driver Version')" \
|
|
||||||
--menu "$menu_text" 28 80 16 \
|
|
||||||
"${choices[@]}" \
|
|
||||||
2>&1 >/dev/tty) || exit 0
|
|
||||||
|
|
||||||
[[ -z "$TARGET_VERSION" ]] && exit 0
|
|
||||||
|
|
||||||
if [[ "$TARGET_VERSION" == "latest" ]]; then
|
|
||||||
TARGET_VERSION="$latest"
|
|
||||||
fi
|
|
||||||
TARGET_VERSION=$(echo "$TARGET_VERSION" | tr -d '[:space:]')
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Overview dialog (current state)
|
|
||||||
# ============================================================
|
|
||||||
show_current_state_dialog() {
|
|
||||||
find_nvidia_containers
|
|
||||||
|
|
||||||
local info
|
|
||||||
info="\n$(translate 'Host NVIDIA driver:') ${HOST_NVIDIA_VERSION}\n"
|
|
||||||
info+="$(translate 'Kernel:') $(uname -r)\n\n"
|
|
||||||
|
|
||||||
if [[ ${#NVIDIA_CONTAINERS[@]} -eq 0 ]]; then
|
|
||||||
info+="$(translate 'No LXC containers with NVIDIA passthrough found.')\n"
|
|
||||||
else
|
|
||||||
info+="$(translate 'LXC containers with NVIDIA passthrough:')\n\n"
|
|
||||||
for ctid in "${NVIDIA_CONTAINERS[@]}"; do
|
|
||||||
local lxc_ver ct_name
|
|
||||||
lxc_ver=$(get_lxc_nvidia_version "$ctid")
|
|
||||||
ct_name=$(pct config "$ctid" 2>/dev/null | grep "^hostname:" | awk '{print $2}')
|
|
||||||
info+=" CT ${ctid} ${ct_name:+(${ct_name})} — $(translate 'driver:') ${lxc_ver}\n"
|
|
||||||
done
|
|
||||||
fi
|
|
||||||
|
|
||||||
info+="\n$(translate 'After selecting a version, LXC containers will be updated first, then the host.')"
|
|
||||||
info+="\n$(translate 'A reboot is required after the host update.')"
|
|
||||||
|
|
||||||
dialog --backtitle "ProxMenux" \
|
|
||||||
--title "$(translate 'NVIDIA Update — Current State')" \
|
|
||||||
--yesno "$info" 20 80 \
|
|
||||||
>/dev/tty 2>&1 || exit 0
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# System preparation (mirror of nvidia_installer.sh)
|
|
||||||
# ============================================================
|
|
||||||
ensure_repos_and_headers() {
|
|
||||||
msg_info "$(translate 'Checking kernel headers and build tools...')"
|
|
||||||
|
|
||||||
local kver
|
|
||||||
kver=$(uname -r)
|
|
||||||
|
|
||||||
apt-get update -qq >>"$LOG_FILE" 2>&1
|
|
||||||
|
|
||||||
if ! dpkg -s "pve-headers-$kver" >/dev/null 2>&1 && \
|
|
||||||
! dpkg -s "proxmox-headers-$kver" >/dev/null 2>&1; then
|
|
||||||
apt-get install -y "pve-headers-$kver" "proxmox-headers-$kver" build-essential dkms >>"$LOG_FILE" 2>&1 || true
|
|
||||||
else
|
|
||||||
apt-get install -y build-essential dkms >>"$LOG_FILE" 2>&1 || true
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "$(translate 'Kernel headers and build tools verified.')" | tee -a "$screen_capture"
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Host NVIDIA cleanup (before update)
|
|
||||||
# ============================================================
|
|
||||||
_stop_nvidia_services() {
|
|
||||||
local services=(
|
|
||||||
"nvidia-persistenced.service"
|
|
||||||
"nvidia-persistenced"
|
|
||||||
"nvidia-powerd.service"
|
|
||||||
)
|
|
||||||
|
|
||||||
local services_detected=0
|
|
||||||
for service in "${services[@]}"; do
|
|
||||||
if systemctl is-active --quiet "$service" 2>/dev/null || \
|
|
||||||
systemctl is-enabled --quiet "$service" 2>/dev/null; then
|
|
||||||
services_detected=1
|
|
||||||
break
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
if [ "$services_detected" -eq 1 ]; then
|
|
||||||
msg_info "$(translate 'Stopping and disabling NVIDIA services...')"
|
|
||||||
for service in "${services[@]}"; do
|
|
||||||
systemctl is-active --quiet "$service" 2>/dev/null && systemctl stop "$service" >/dev/null 2>&1 || true
|
|
||||||
systemctl is-enabled --quiet "$service" 2>/dev/null && systemctl disable "$service" >/dev/null 2>&1 || true
|
|
||||||
done
|
|
||||||
sleep 2
|
|
||||||
msg_ok "$(translate 'NVIDIA services stopped and disabled.')" | tee -a "$screen_capture"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
_unload_nvidia_modules() {
|
|
||||||
msg_info "$(translate 'Unloading NVIDIA kernel modules...')"
|
|
||||||
|
|
||||||
for mod in nvidia_uvm nvidia_drm nvidia_modeset nvidia; do
|
|
||||||
modprobe -r "$mod" >/dev/null 2>&1 || true
|
|
||||||
done
|
|
||||||
|
|
||||||
if lsmod | grep -qi '\bnvidia'; then
|
|
||||||
for mod in nvidia_uvm nvidia_drm nvidia_modeset nvidia; do
|
|
||||||
modprobe -r --force "$mod" >/dev/null 2>&1 || true
|
|
||||||
done
|
|
||||||
fi
|
|
||||||
|
|
||||||
if lsmod | grep -qi '\bnvidia'; then
|
|
||||||
msg_warn "$(translate 'Some NVIDIA modules could not be unloaded. Update may fail. Ensure no processes are using the GPU.')"
|
|
||||||
else
|
|
||||||
msg_ok "$(translate 'NVIDIA kernel modules unloaded successfully.')" | tee -a "$screen_capture"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
cleanup_nvidia_dkms() {
|
|
||||||
local versions
|
|
||||||
versions=$(dkms status 2>/dev/null | awk -F, '/nvidia/ {gsub(/ /,"",$2); print $2}' || true)
|
|
||||||
[[ -z "$versions" ]] && return 0
|
|
||||||
|
|
||||||
msg_info "$(translate 'Removing NVIDIA DKMS entries...')"
|
|
||||||
while IFS= read -r ver; do
|
|
||||||
[[ -z "$ver" ]] && continue
|
|
||||||
dkms remove -m nvidia -v "$ver" --all >/dev/null 2>&1 || true
|
|
||||||
done <<< "$versions"
|
|
||||||
msg_ok "$(translate 'NVIDIA DKMS entries removed.')" | tee -a "$screen_capture"
|
|
||||||
}
|
|
||||||
|
|
||||||
_purge_nvidia_host() {
|
|
||||||
msg_info2 "$(translate 'Preparing host for driver update...')"
|
|
||||||
|
|
||||||
_stop_nvidia_services
|
|
||||||
_unload_nvidia_modules
|
|
||||||
|
|
||||||
if command -v nvidia-uninstall >/dev/null 2>&1; then
|
|
||||||
msg_info "$(translate 'Running nvidia-uninstall...')"
|
|
||||||
nvidia-uninstall --silent >>"$LOG_FILE" 2>&1 || true
|
|
||||||
msg_ok "$(translate 'nvidia-uninstall completed.')" | tee -a "$screen_capture"
|
|
||||||
fi
|
|
||||||
|
|
||||||
cleanup_nvidia_dkms
|
|
||||||
|
|
||||||
msg_info "$(translate 'Purging NVIDIA packages...')"
|
|
||||||
apt-get -y purge 'nvidia-*' 'libnvidia-*' 'cuda-*' 'libcudnn*' >>"$LOG_FILE" 2>&1 || true
|
|
||||||
apt-get -y autoremove --purge >>"$LOG_FILE" 2>&1 || true
|
|
||||||
msg_ok "$(translate 'NVIDIA packages purged.')" | tee -a "$screen_capture"
|
|
||||||
|
|
||||||
# Remove stale udev / modprobe files so the new installer can write fresh ones
|
|
||||||
rm -f /etc/udev/rules.d/70-nvidia.rules
|
|
||||||
rm -f /etc/modprobe.d/nvidia*.conf /usr/lib/modprobe.d/nvidia*.conf
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Download installer (with integrity check — mirror of installer)
|
|
||||||
# ============================================================
|
|
||||||
ensure_workdir() {
|
|
||||||
mkdir -p "$NVIDIA_WORKDIR"
|
|
||||||
}
|
|
||||||
|
|
||||||
verify_version_exists() {
|
|
||||||
local version="$1"
|
|
||||||
local url="${NVIDIA_BASE_URL}/${version}/"
|
|
||||||
if curl -fsSL --head "$url" >/dev/null 2>&1; then
|
|
||||||
return 0
|
|
||||||
else
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
download_nvidia_installer() {
|
|
||||||
ensure_workdir
|
|
||||||
local version="$1"
|
|
||||||
version=$(echo "$version" | tr -d '[:space:]' | tr -d '\n' | tr -d '\r')
|
|
||||||
|
|
||||||
if [[ ! "$version" =~ ^[0-9]+\.[0-9]+(\.[0-9]+)?$ ]]; then
|
|
||||||
msg_error "Invalid version format: $version" >&2
|
|
||||||
echo "ERROR: Invalid version format: '$version'" >> "$LOG_FILE"
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
local run_file="$NVIDIA_WORKDIR/NVIDIA-Linux-x86_64-${version}.run"
|
|
||||||
|
|
||||||
if [[ -f "$run_file" ]]; then
|
|
||||||
local existing_size file_type
|
|
||||||
existing_size=$(stat -c%s "$run_file" 2>/dev/null || echo "0")
|
|
||||||
file_type=$(file "$run_file" 2>/dev/null || echo "unknown")
|
|
||||||
|
|
||||||
if [[ $existing_size -gt 40000000 ]] && echo "$file_type" | grep -q "executable"; then
|
|
||||||
if sh "$run_file" --check 2>&1 | tee -a "$LOG_FILE" | grep -q "OK"; then
|
|
||||||
msg_ok "$(translate 'Installer already downloaded and verified.')" | tee -a "$screen_capture" >&2
|
|
||||||
printf '%s\n' "$run_file"
|
|
||||||
return 0
|
|
||||||
else
|
|
||||||
msg_warn "$(translate 'Existing file failed verification, re-downloading...')" >&2
|
|
||||||
rm -f "$run_file"
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
msg_warn "$(translate 'Removing invalid existing file...')" >&2
|
|
||||||
rm -f "$run_file"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! verify_version_exists "$version"; then
|
|
||||||
msg_error "Version $version does not exist on NVIDIA servers" >&2
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
local urls=(
|
|
||||||
"${NVIDIA_BASE_URL}/${version}/NVIDIA-Linux-x86_64-${version}.run"
|
|
||||||
"${NVIDIA_BASE_URL}/${version}/NVIDIA-Linux-x86_64-${version}-no-compat32.run"
|
|
||||||
)
|
|
||||||
|
|
||||||
# Header line on the real terminal so it stays visible regardless of caller redirects.
|
|
||||||
printf '\n %s NVIDIA-Linux-x86_64-%s.run\n' \
|
|
||||||
"$(translate 'Downloading')" "$version" >/dev/tty
|
|
||||||
|
|
||||||
local success=false
|
|
||||||
for url in "${urls[@]}"; do
|
|
||||||
rm -f "$run_file"
|
|
||||||
echo "Attempting download from: $url" >> "$LOG_FILE"
|
|
||||||
|
|
||||||
# wget --show-progress writes its progress bar to stderr. We route it to
|
|
||||||
# /dev/tty explicitly so the user always sees it (same UX as ISO downloads
|
|
||||||
# in vm_creator.sh). The file contents still go to $run_file.
|
|
||||||
if wget --no-verbose --show-progress \
|
|
||||||
--connect-timeout=30 --timeout=600 --tries=1 \
|
|
||||||
-O "$run_file" "$url" 2>/dev/tty; then
|
|
||||||
[[ ! -f "$run_file" ]] && { echo "ERROR: File not created" >> "$LOG_FILE"; continue; }
|
|
||||||
local file_size file_type
|
|
||||||
file_size=$(stat -c%s "$run_file" 2>/dev/null || echo "0")
|
|
||||||
file_type=$(file "$run_file" 2>/dev/null)
|
|
||||||
echo "Downloaded file size: $file_size bytes, type: $file_type" >> "$LOG_FILE"
|
|
||||||
if [[ $file_size -gt 40000000 ]] && echo "$file_type" | grep -q "executable"; then
|
|
||||||
success=true
|
|
||||||
break
|
|
||||||
fi
|
|
||||||
rm -f "$run_file"
|
|
||||||
else
|
|
||||||
echo "ERROR: wget failed for $url (exit: $?)" >> "$LOG_FILE"
|
|
||||||
rm -f "$run_file"
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
if ! $success; then
|
|
||||||
msg_error "$(translate 'Download failed. Check') ${LOG_FILE}" >&2
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
chmod +x "$run_file"
|
|
||||||
msg_ok "$(translate 'Download complete.')" | tee -a "$screen_capture" >&2
|
|
||||||
printf '%s\n' "$run_file"
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Host installer run (visible output — mirror of installer)
|
|
||||||
# ============================================================
|
|
||||||
run_host_installer() {
|
|
||||||
local installer="$1"
|
|
||||||
local tmp_extract_dir="$NVIDIA_WORKDIR/tmp_extract"
|
|
||||||
mkdir -p "$tmp_extract_dir"
|
|
||||||
|
|
||||||
msg_info2 "$(translate 'Starting NVIDIA installer on host. This may take several minutes...')"
|
|
||||||
echo "" >>"$LOG_FILE"
|
|
||||||
echo "=== Running NVIDIA installer: $installer ===" >>"$LOG_FILE"
|
|
||||||
|
|
||||||
sh "$installer" \
|
|
||||||
--tmpdir="$tmp_extract_dir" \
|
|
||||||
--no-questions \
|
|
||||||
--ui=none \
|
|
||||||
--disable-nouveau \
|
|
||||||
--no-nouveau-check \
|
|
||||||
--dkms \
|
|
||||||
2>&1 | tee -a "$LOG_FILE"
|
|
||||||
local rc=${PIPESTATUS[0]}
|
|
||||||
echo "" >>"$LOG_FILE"
|
|
||||||
|
|
||||||
rm -rf "$tmp_extract_dir"
|
|
||||||
|
|
||||||
if [[ $rc -ne 0 ]]; then
|
|
||||||
msg_error "$(translate 'NVIDIA installer reported an error. Check') ${LOG_FILE}"
|
|
||||||
update_component_status "nvidia_driver" "failed" "" "gpu" '{"patched":false}'
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "$(translate 'NVIDIA driver installed on host.')" | tee -a "$screen_capture"
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# LXC NVIDIA update — aligned with add_gpu_lxc.sh::_install_nvidia_drivers
|
|
||||||
# ============================================================
|
|
||||||
CT_ORIG_MEM=""
|
|
||||||
NVIDIA_INSTALL_MIN_MB=2048
|
|
||||||
CT_WAS_STARTED_FOR_UPDATE=false
|
|
||||||
|
|
||||||
_detect_container_distro() {
|
|
||||||
local distro
|
|
||||||
distro=$(pct exec "$1" -- grep "^ID=" /etc/os-release 2>/dev/null \
|
|
||||||
| cut -d= -f2 | tr -d '[:space:]"')
|
|
||||||
echo "${distro:-unknown}"
|
|
||||||
}
|
|
||||||
|
|
||||||
_ensure_container_memory() {
|
|
||||||
local ctid="$1"
|
|
||||||
local cur_mem
|
|
||||||
cur_mem=$(pct config "$ctid" 2>/dev/null | awk '/^memory:/{print $2}')
|
|
||||||
[[ -z "$cur_mem" ]] && cur_mem=512
|
|
||||||
|
|
||||||
if [[ "$cur_mem" -lt "$NVIDIA_INSTALL_MIN_MB" ]]; then
|
|
||||||
if whiptail --title "$(translate 'Low Container Memory')" --yesno \
|
|
||||||
"$(translate 'Container') ${ctid} $(translate 'has') ${cur_mem}MB RAM.\n\n$(translate 'The NVIDIA installer needs at least') ${NVIDIA_INSTALL_MIN_MB}MB $(translate 'to run without being killed by the OOM killer.')\n\n$(translate 'Increase container RAM temporarily to') ${NVIDIA_INSTALL_MIN_MB}MB?" \
|
|
||||||
13 72; then
|
|
||||||
CT_ORIG_MEM="$cur_mem"
|
|
||||||
pct set "$ctid" -memory "$NVIDIA_INSTALL_MIN_MB" >>"$LOG_FILE" 2>&1 || true
|
|
||||||
else
|
|
||||||
msg_warn "$(translate 'Insufficient memory. Skipping LXC') ${ctid}."
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
_restore_container_memory() {
|
|
||||||
local ctid="$1"
|
|
||||||
if [[ -n "$CT_ORIG_MEM" ]]; then
|
|
||||||
msg_info "$(translate 'Restoring container memory to') ${CT_ORIG_MEM}MB..."
|
|
||||||
pct set "$ctid" -memory "$CT_ORIG_MEM" >>"$LOG_FILE" 2>&1 || true
|
|
||||||
msg_ok "$(translate 'Memory restored.')"
|
|
||||||
CT_ORIG_MEM=""
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
start_container_and_wait() {
|
|
||||||
local ctid="$1"
|
|
||||||
msg_info "$(translate 'Starting container') ${ctid}..."
|
|
||||||
pct start "$ctid" >>"$LOG_FILE" 2>&1 || true
|
|
||||||
|
|
||||||
local ready=false
|
|
||||||
for _ in {1..15}; do
|
|
||||||
sleep 2
|
|
||||||
if pct exec "$ctid" -- true >/dev/null 2>&1; then
|
|
||||||
ready=true
|
|
||||||
break
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
if ! $ready; then
|
|
||||||
msg_warn "$(translate 'Container') ${ctid} $(translate 'did not become ready. Skipping.')"
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
msg_ok "$(translate 'Container') ${ctid} $(translate 'started.')" | tee -a "$screen_capture"
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
update_lxc_nvidia() {
|
|
||||||
local ctid="$1"
|
|
||||||
local version="$2"
|
|
||||||
CT_WAS_STARTED_FOR_UPDATE=false
|
|
||||||
|
|
||||||
local old_version
|
|
||||||
old_version=$(get_lxc_nvidia_version "$ctid")
|
|
||||||
|
|
||||||
msg_info2 "$(translate 'Container') ${ctid}: $(translate 'updating NVIDIA userspace libs') (${old_version} → ${version})"
|
|
||||||
|
|
||||||
# Start the container if stopped (required for pct exec based install)
|
|
||||||
if ! pct status "$ctid" 2>/dev/null | grep -q "running"; then
|
|
||||||
CT_WAS_STARTED_FOR_UPDATE=true
|
|
||||||
if ! start_container_and_wait "$ctid"; then
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Detect distro (alpine / arch / debian-like)
|
|
||||||
msg_info "$(translate 'Detecting container OS...')"
|
|
||||||
local distro
|
|
||||||
distro=$(_detect_container_distro "$ctid")
|
|
||||||
msg_ok "$(translate 'Container OS:') ${distro}" | tee -a "$screen_capture"
|
|
||||||
|
|
||||||
local install_rc=0
|
|
||||||
|
|
||||||
case "$distro" in
|
|
||||||
alpine)
|
|
||||||
# Alpine: musl — use apk nvidia-utils (repo-managed, no .run)
|
|
||||||
msg_info2 "$(translate 'Upgrading NVIDIA utils (Alpine)...')"
|
|
||||||
pct exec "$ctid" -- sh -c \
|
|
||||||
"apk update && apk add --no-cache --upgrade nvidia-utils" \
|
|
||||||
2>&1 | tee -a "$LOG_FILE"
|
|
||||||
install_rc=${PIPESTATUS[0]}
|
|
||||||
;;
|
|
||||||
|
|
||||||
arch|manjaro|endeavouros)
|
|
||||||
msg_info2 "$(translate 'Upgrading NVIDIA utils (Arch)...')"
|
|
||||||
pct exec "$ctid" -- bash -c \
|
|
||||||
"pacman -Syu --noconfirm nvidia-utils" \
|
|
||||||
2>&1 | tee -a "$LOG_FILE"
|
|
||||||
install_rc=${PIPESTATUS[0]}
|
|
||||||
;;
|
|
||||||
|
|
||||||
*)
|
|
||||||
# Debian / Ubuntu / generic glibc: use the host-cached .run binary
|
|
||||||
local run_file="${NVIDIA_WORKDIR}/NVIDIA-Linux-x86_64-${version}.run"
|
|
||||||
|
|
||||||
if [[ ! -f "$run_file" ]]; then
|
|
||||||
msg_warn "$(translate 'Installer not found:') ${run_file}. $(translate 'Skipping LXC') ${ctid}."
|
|
||||||
install_rc=1
|
|
||||||
else
|
|
||||||
# Memory check — nvidia-installer needs ~2GB during install
|
|
||||||
if ! _ensure_container_memory "$ctid"; then
|
|
||||||
install_rc=1
|
|
||||||
else
|
|
||||||
# Disk space check — NVIDIA libs need ~1.5 GB free in the container
|
|
||||||
local free_mb
|
|
||||||
free_mb=$(pct exec "$ctid" -- df -m / 2>/dev/null | awk 'NR==2{print $4}' || echo 0)
|
|
||||||
if [[ "$free_mb" -lt 1500 ]]; then
|
|
||||||
_restore_container_memory "$ctid"
|
|
||||||
dialog --backtitle "ProxMenux" \
|
|
||||||
--title "$(translate 'Insufficient Disk Space')" \
|
|
||||||
--msgbox "\n$(translate 'Container') ${ctid} $(translate 'has only') ${free_mb}MB $(translate 'of free disk space.')\n\n$(translate 'NVIDIA libs require approximately 1.5GB of free space.')\n\n$(translate 'Please expand the container disk and run this option again.')" \
|
|
||||||
12 72
|
|
||||||
msg_warn "$(translate 'Insufficient disk space. Skipping LXC') ${ctid}."
|
|
||||||
install_rc=1
|
|
||||||
else
|
|
||||||
# Extract .run on the host (avoids decompression OOM inside container)
|
|
||||||
local extract_dir="${NVIDIA_WORKDIR}/extracted_${version}"
|
|
||||||
local archive="/tmp/nvidia_lxc_${version}.tar.gz"
|
|
||||||
|
|
||||||
msg_info2 "$(translate 'Extracting NVIDIA installer on host...')"
|
|
||||||
rm -rf "$extract_dir"
|
|
||||||
sh "$run_file" --extract-only --target "$extract_dir" 2>&1 | tee -a "$LOG_FILE"
|
|
||||||
if [[ ${PIPESTATUS[0]} -ne 0 ]]; then
|
|
||||||
msg_warn "$(translate 'Extraction failed. Check log:') ${LOG_FILE}"
|
|
||||||
_restore_container_memory "$ctid"
|
|
||||||
install_rc=1
|
|
||||||
else
|
|
||||||
msg_ok "$(translate 'NVIDIA installer extracted.')" | tee -a "$screen_capture"
|
|
||||||
|
|
||||||
msg_info2 "$(translate 'Packing installer archive...')"
|
|
||||||
tar --checkpoint=5000 --checkpoint-action=dot \
|
|
||||||
-czf "$archive" -C "$extract_dir" . 2>&1 | tee -a "$LOG_FILE"
|
|
||||||
echo ""
|
|
||||||
local archive_size
|
|
||||||
archive_size=$(du -sh "$archive" 2>/dev/null | cut -f1)
|
|
||||||
msg_ok "$(translate 'Archive ready') (${archive_size})." | tee -a "$screen_capture"
|
|
||||||
|
|
||||||
msg_info "$(translate 'Copying installer to container') ${ctid}..."
|
|
||||||
if ! pct push "$ctid" "$archive" /tmp/nvidia_lxc.tar.gz >>"$LOG_FILE" 2>&1; then
|
|
||||||
msg_warn "$(translate 'pct push failed. Check log:') ${LOG_FILE}"
|
|
||||||
rm -f "$archive"
|
|
||||||
rm -rf "$extract_dir"
|
|
||||||
_restore_container_memory "$ctid"
|
|
||||||
install_rc=1
|
|
||||||
else
|
|
||||||
rm -f "$archive"
|
|
||||||
msg_ok "$(translate 'Installer copied to container.')" | tee -a "$screen_capture"
|
|
||||||
|
|
||||||
msg_info2 "$(translate 'Running NVIDIA installer in container. This may take several minutes...')"
|
|
||||||
echo "" >>"$LOG_FILE"
|
|
||||||
pct exec "$ctid" -- bash -c "
|
|
||||||
mkdir -p /tmp/nvidia_lxc_install
|
|
||||||
tar -xzf /tmp/nvidia_lxc.tar.gz -C /tmp/nvidia_lxc_install 2>&1
|
|
||||||
/tmp/nvidia_lxc_install/nvidia-installer \
|
|
||||||
--no-kernel-modules \
|
|
||||||
--no-questions \
|
|
||||||
--ui=none \
|
|
||||||
--no-nouveau-check \
|
|
||||||
--no-dkms \
|
|
||||||
--no-install-compat32-libs
|
|
||||||
EXIT=\$?
|
|
||||||
rm -rf /tmp/nvidia_lxc_install /tmp/nvidia_lxc.tar.gz
|
|
||||||
exit \$EXIT
|
|
||||||
" 2>&1 | tee -a "$LOG_FILE"
|
|
||||||
install_rc=${PIPESTATUS[0]}
|
|
||||||
|
|
||||||
rm -rf "$extract_dir"
|
|
||||||
_restore_container_memory "$ctid"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
|
|
||||||
if [[ $install_rc -ne 0 ]]; then
|
|
||||||
msg_warn "$(translate 'NVIDIA update failed for LXC') ${ctid} ($(translate 'rc='))${install_rc}. $(translate 'Check log:') ${LOG_FILE}"
|
|
||||||
if [[ "$CT_WAS_STARTED_FOR_UPDATE" == "true" ]]; then
|
|
||||||
msg_info "$(translate 'Stopping container') ${ctid}..."
|
|
||||||
pct stop "$ctid" >>"$LOG_FILE" 2>&1 || true
|
|
||||||
msg_ok "$(translate 'Container stopped.')" | tee -a "$screen_capture"
|
|
||||||
fi
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Verify nvidia-smi inside the container
|
|
||||||
if pct exec "$ctid" -- sh -c "which nvidia-smi" >/dev/null 2>&1; then
|
|
||||||
local new_ver
|
|
||||||
new_ver=$(pct exec "$ctid" -- nvidia-smi \
|
|
||||||
--query-gpu=driver_version --format=csv,noheader 2>/dev/null \
|
|
||||||
| head -1 | tr -d '[:space:]' || true)
|
|
||||||
msg_ok "$(translate 'Container') ${ctid}: ${old_version} → ${new_ver:-$version}" | tee -a "$screen_capture"
|
|
||||||
else
|
|
||||||
msg_warn "$(translate 'nvidia-smi not found in container') ${ctid} $(translate 'after update.')"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ "$CT_WAS_STARTED_FOR_UPDATE" == "true" ]]; then
|
|
||||||
msg_info "$(translate 'Stopping container') ${ctid}..."
|
|
||||||
pct stop "$ctid" >>"$LOG_FILE" 2>&1 || true
|
|
||||||
msg_ok "$(translate 'Container stopped.')" | tee -a "$screen_capture"
|
|
||||||
fi
|
|
||||||
return 0
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Restart prompt
|
|
||||||
# ============================================================
|
|
||||||
restart_prompt() {
|
|
||||||
echo
|
|
||||||
msg_info "$(translate 'Removing no longer required packages and purging old cached updates...')"
|
|
||||||
apt-get -y autoremove >/dev/null 2>&1
|
|
||||||
apt-get -y autoclean >/dev/null 2>&1
|
|
||||||
msg_ok "$(translate 'Cleanup finished.')" | tee -a "$screen_capture"
|
|
||||||
|
|
||||||
if whiptail --title "$(translate 'Reboot Required')" \
|
|
||||||
--yesno "$(translate 'The host driver update requires a reboot to take effect. Do you want to restart now?')" 10 70; then
|
|
||||||
msg_success "$(translate 'Press Enter to continue...')"
|
|
||||||
read -r
|
|
||||||
msg_warn "$(translate 'Rebooting the system...')"
|
|
||||||
rm -f "$screen_capture"
|
|
||||||
reboot
|
|
||||||
else
|
|
||||||
msg_info2 "$(translate 'You can reboot later manually.')"
|
|
||||||
msg_success "$(translate 'Press Enter to continue...')"
|
|
||||||
read -r
|
|
||||||
rm -f "$screen_capture"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
# ============================================================
|
|
||||||
# Main
|
|
||||||
# ============================================================
|
|
||||||
main() {
|
|
||||||
: >"$LOG_FILE"
|
|
||||||
: >"$screen_capture"
|
|
||||||
|
|
||||||
# ---- Phase 1: dialogs ----
|
|
||||||
check_gpu_not_in_vm_passthrough
|
|
||||||
detect_host_nvidia
|
|
||||||
get_kernel_compatibility_info
|
|
||||||
show_current_state_dialog
|
|
||||||
select_target_version
|
|
||||||
|
|
||||||
# Same version confirmation
|
|
||||||
if [[ "$TARGET_VERSION" == "$HOST_NVIDIA_VERSION" ]]; then
|
|
||||||
if ! dialog --backtitle "ProxMenux" \
|
|
||||||
--title "$(translate 'Same Version')" \
|
|
||||||
--yesno "\n$(translate 'Version') ${TARGET_VERSION} $(translate 'is already installed on the host.')\n\n$(translate 'Reinstall and force-update all LXC containers anyway?')" \
|
|
||||||
10 70 >/dev/tty 2>&1; then
|
|
||||||
exit 0
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
# ---- Phase 2: processing ----
|
|
||||||
show_proxmenux_logo
|
|
||||||
msg_title "$(translate "$SCRIPT_TITLE")"
|
|
||||||
|
|
||||||
ensure_repos_and_headers
|
|
||||||
|
|
||||||
# Download installer once — shared between LXC and host updates.
|
|
||||||
# No 2>>"$LOG_FILE" redirect: we want msg_warn/msg_error from the function to
|
|
||||||
# reach the user's terminal, and wget's progress bar goes to /dev/tty directly.
|
|
||||||
local installer
|
|
||||||
installer=$(download_nvidia_installer "$TARGET_VERSION")
|
|
||||||
local download_result=$?
|
|
||||||
|
|
||||||
if [[ $download_result -ne 0 || -z "$installer" || ! -f "$installer" ]]; then
|
|
||||||
msg_error "$(translate 'Failed to obtain NVIDIA installer. Check') ${LOG_FILE}"
|
|
||||||
rm -f "$screen_capture"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Update LXCs first (userspace libs only — doesn't need a reboot)
|
|
||||||
if [[ ${#NVIDIA_CONTAINERS[@]} -gt 0 ]]; then
|
|
||||||
msg_info2 "$(translate 'Updating LXC containers...')"
|
|
||||||
for ctid in "${NVIDIA_CONTAINERS[@]}"; do
|
|
||||||
update_lxc_nvidia "$ctid" "$TARGET_VERSION" || true
|
|
||||||
done
|
|
||||||
fi
|
|
||||||
|
|
||||||
# Purge and reinstall host driver
|
|
||||||
_purge_nvidia_host
|
|
||||||
|
|
||||||
if ! run_host_installer "$installer"; then
|
|
||||||
rm -f "$screen_capture"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_info "$(translate 'Updating initramfs for all kernels...')"
|
|
||||||
update-initramfs -u -k all >>"$LOG_FILE" 2>&1 || true
|
|
||||||
msg_ok "$(translate 'initramfs updated.')" | tee -a "$screen_capture"
|
|
||||||
|
|
||||||
# ---- Phase 3: summary ----
|
|
||||||
sleep 2
|
|
||||||
show_proxmenux_logo
|
|
||||||
msg_title "$(translate "$SCRIPT_TITLE")"
|
|
||||||
cat "$screen_capture"
|
|
||||||
echo -e "${TAB}${GN}📄 $(translate "Log file")${CL}: ${BL}$LOG_FILE${CL}"
|
|
||||||
|
|
||||||
msg_info2 "$(translate 'Checking NVIDIA driver status with nvidia-smi')"
|
|
||||||
if command -v nvidia-smi >/dev/null 2>&1; then
|
|
||||||
nvidia-smi || true
|
|
||||||
local NEW_HOST_VERSION
|
|
||||||
NEW_HOST_VERSION=$(nvidia-smi --query-gpu=driver_version --format=csv,noheader 2>/dev/null | head -n1)
|
|
||||||
if [[ -n "$NEW_HOST_VERSION" ]]; then
|
|
||||||
msg_ok "$(translate 'NVIDIA driver') $NEW_HOST_VERSION $(translate 'installed successfully on host.')"
|
|
||||||
update_component_status "nvidia_driver" "installed" "$NEW_HOST_VERSION" "gpu" '{"patched":false}'
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
msg_warn "$(translate 'nvidia-smi not found in PATH. Verify the update manually after reboot.')"
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_success "$(translate 'NVIDIA driver update completed.')"
|
|
||||||
restart_prompt
|
|
||||||
}
|
|
||||||
|
|
||||||
main
|
|
||||||
@@ -27,24 +27,23 @@ initialize_cache
|
|||||||
while true; do
|
while true; do
|
||||||
OPTION=$(dialog --colors --backtitle "ProxMenux" \
|
OPTION=$(dialog --colors --backtitle "ProxMenux" \
|
||||||
--title "$(translate "GPUs and Coral-TPU Menu")" \
|
--title "$(translate "GPUs and Coral-TPU Menu")" \
|
||||||
--menu "\n$(translate "Select an option:")" 26 78 18 \
|
--menu "\n$(translate "Select an option:")" 24 78 16 \
|
||||||
"" "\Z4──────────────────────── HOST ─────────────────────────\Zn" \
|
"" "\Z4──────────────────────── HOST ─────────────────────────\Zn" \
|
||||||
"1" "$(translate "Install NVIDIA Drivers on Host")" \
|
"1" "$(translate "Install/Update NVIDIA Drivers (Host + LXC)")" \
|
||||||
"2" "$(translate "Update NVIDIA Drivers (Host + LXC)")" \
|
"2" "$(translate "Install/Update Coral TPU on Host")" \
|
||||||
"3" "$(translate "Install/Update Coral TPU on Host")" \
|
|
||||||
"" "" \
|
"" "" \
|
||||||
"" "\Z4──────────────────────── LXC ──────────────────────────\Zn" \
|
"" "\Z4──────────────────────── LXC ──────────────────────────\Zn" \
|
||||||
"4" "$(translate "Add GPU to LXC (Intel | AMD | NVIDIA)") \Zb\Z4Switch Mode\Zn" \
|
"3" "$(translate "Add GPU to LXC (Intel | AMD | NVIDIA)") \Zb\Z4Switch Mode\Zn" \
|
||||||
"5" "$(translate "Add Coral TPU to LXC")" \
|
"4" "$(translate "Add Coral TPU to LXC")" \
|
||||||
"" "" \
|
"" "" \
|
||||||
"" "\Z4──────────────────────── VM ───────────────────────────\Zn" \
|
"" "\Z4──────────────────────── VM ───────────────────────────\Zn" \
|
||||||
"6" "$(translate "Add GPU to VM (Intel | AMD | NVIDIA)") \Zb\Z4Switch Mode\Zn" \
|
"5" "$(translate "Add GPU to VM (Intel | AMD | NVIDIA)") \Zb\Z4Switch Mode\Zn" \
|
||||||
"" "" \
|
"" "" \
|
||||||
"" "\Z4──────────────────── SWICHT MODE ───────────────────────\Zn" \
|
"" "\Z4──────────────────── SWICHT MODE ───────────────────────\Zn" \
|
||||||
"7" "$(translate "Switch GPU Mode (VM <-> LXC)")" \
|
"6" "$(translate "Switch GPU Mode (VM <-> LXC)")" \
|
||||||
"" "" \
|
"" "" \
|
||||||
"" "\Z4────────────────────── Utilities ───────────────────────\Zn" \
|
"" "\Z4────────────────────── Utilities ───────────────────────\Zn" \
|
||||||
"8" "$(translate "Manual CLI Guide (GPU/TPU)")" \
|
"7" "$(translate "Manual CLI Guide (GPU/TPU)")" \
|
||||||
"0" "$(translate "Return to Main Menu")" \
|
"0" "$(translate "Return to Main Menu")" \
|
||||||
2>&1 >/dev/tty
|
2>&1 >/dev/tty
|
||||||
) || { exec bash "$LOCAL_SCRIPTS/menus/main_menu.sh"; }
|
) || { exec bash "$LOCAL_SCRIPTS/menus/main_menu.sh"; }
|
||||||
@@ -54,24 +53,21 @@ while true; do
|
|||||||
bash "$LOCAL_SCRIPTS/gpu_tpu/nvidia_installer.sh"
|
bash "$LOCAL_SCRIPTS/gpu_tpu/nvidia_installer.sh"
|
||||||
;;
|
;;
|
||||||
2)
|
2)
|
||||||
bash "$LOCAL_SCRIPTS/gpu_tpu/nvidia_update.sh"
|
|
||||||
;;
|
|
||||||
3)
|
|
||||||
bash "$LOCAL_SCRIPTS/gpu_tpu/install_coral_pve9.sh"
|
bash "$LOCAL_SCRIPTS/gpu_tpu/install_coral_pve9.sh"
|
||||||
;;
|
;;
|
||||||
4)
|
3)
|
||||||
bash "$LOCAL_SCRIPTS/gpu_tpu/add_gpu_lxc.sh"
|
bash "$LOCAL_SCRIPTS/gpu_tpu/add_gpu_lxc.sh"
|
||||||
;;
|
;;
|
||||||
5)
|
4)
|
||||||
bash "$LOCAL_SCRIPTS/gpu_tpu/install_coral_lxc.sh"
|
bash "$LOCAL_SCRIPTS/gpu_tpu/install_coral_lxc.sh"
|
||||||
;;
|
;;
|
||||||
6)
|
5)
|
||||||
bash "$LOCAL_SCRIPTS/gpu_tpu/add_gpu_vm.sh"
|
bash "$LOCAL_SCRIPTS/gpu_tpu/add_gpu_vm.sh"
|
||||||
;;
|
;;
|
||||||
7)
|
6)
|
||||||
bash "$LOCAL_SCRIPTS/gpu_tpu/switch_gpu_mode.sh"
|
bash "$LOCAL_SCRIPTS/gpu_tpu/switch_gpu_mode.sh"
|
||||||
;;
|
;;
|
||||||
8)
|
7)
|
||||||
bash "$LOCAL_SCRIPTS/gpu_tpu/gpu-tpu-manual-guide.sh"
|
bash "$LOCAL_SCRIPTS/gpu_tpu/gpu-tpu-manual-guide.sh"
|
||||||
;;
|
;;
|
||||||
0)
|
0)
|
||||||
|
|||||||
Reference in New Issue
Block a user