From 012b156b46e472cb0f1d177a5d2f5075f470c3c8 Mon Sep 17 00:00:00 2001 From: MacRimi Date: Fri, 26 Sep 2025 00:17:32 +0200 Subject: [PATCH] Update install_coral_pve9.sh --- scripts/gpu_tpu/install_coral_pve9.sh | 151 +++++++++++++++----------- 1 file changed, 88 insertions(+), 63 deletions(-) diff --git a/scripts/gpu_tpu/install_coral_pve9.sh b/scripts/gpu_tpu/install_coral_pve9.sh index 2921a5e..ead4496 100644 --- a/scripts/gpu_tpu/install_coral_pve9.sh +++ b/scripts/gpu_tpu/install_coral_pve9.sh @@ -1,103 +1,128 @@ #!/bin/bash - # ProxMenux - Coral TPU Installer (PVE 9.x) # ========================================= # Author : MacRimi # License : MIT -# Version : 1.1 (PVE9) +# Version : 1.3 (PVE9, silent build) # Last Updated: 25/09/2025 # ========================================= REPO_URL="https://raw.githubusercontent.com/MacRimi/ProxMenux/main" BASE_DIR="/usr/local/share/proxmenux" UTILS_FILE="$BASE_DIR/utils.sh" +LOG_FILE="/tmp/coral_install.log" if [[ -f "$UTILS_FILE" ]]; then - source "$UTILS_FILE" + source "$UTILS_FILE" fi + load_language initialize_cache -# Prompt before installation pre_install_prompt() { - if ! dialog --title "$(translate 'Coral TPU Installation')" --yesno "$(translate 'Installing Coral TPU drivers requires rebooting the server after installation. Do you want to proceed?')" 10 70; then - msg_warn "$(translate 'Installation cancelled by user.')" - exit 0 - fi + if ! dialog --title "$(translate 'Coral TPU Installation')" --yesno \ + "\n$(translate 'Installing Coral TPU drivers requires rebooting the server after installation. Do you want to proceed?')" 10 70; then + + exit 0 + fi } -# Function to install Coral TPU drivers on PVE 9 install_coral_host() { - show_proxmenux_logo + show_proxmenux_logo + : >"$LOG_FILE" - msg_info "$(translate 'Installing build dependencies...')" - apt-get update -qq - apt-get install -y git devscripts dh-dkms dkms proxmox-headers-$(uname -r) >/dev/null 2>&1 - cd /tmp - rm -rf gasket-driver - git clone https://github.com/google/gasket-driver.git - if [ $? -ne 0 ]; then - msg_error "$(translate 'Error: Could not clone the repository.')" - exit 1 - fi - cd gasket-driver/ + msg_info "$(translate 'Installing build dependencies...')" + apt-get update -qq >>"$LOG_FILE" 2>&1 + apt-get install -y git devscripts dh-dkms dkms proxmox-headers-$(uname -r) >>"$LOG_FILE" 2>&1 + if [[ $? -ne 0 ]]; then msg_error "$(translate 'Error installing build dependencies. Check /tmp/coral_install.log')"; exit 1; fi + msg_ok "$(translate 'Build dependencies installed.')" - # === Apply kernel compatibility patches === - msg_info "$(translate 'Patching source for kernel compatibility...')" - # Fix no_llseek -> noop_llseek - sed -i 's/\.llseek = no_llseek/\.llseek = noop_llseek/' src/gasket_core.c - # Fix DMA_BUF namespace (with quotes) - sed -i 's/^MODULE_IMPORT_NS(DMA_BUF);/MODULE_IMPORT_NS("DMA_BUF");/' src/gasket_page_table.c - # Patch debian/control for Proxmox headers - sed -i "s/linux-headers-686-pae | linux-headers-amd64 | linux-headers-generic | linux-headers/linux-headers-686-pae | linux-headers-amd64 | linux-headers-generic | linux-headers | proxmox-headers-$(uname -r) | pve-headers-$(uname -r)/" debian/control - # === Build DKMS package === - msg_info "$(translate 'Building DKMS package...')" - debuild -us -uc -tc -b - if [ $? -ne 0 ]; then - msg_error "$(translate 'Error: Failed to build driver packages.')" - exit 1 - fi - # === Install DKMS package === - msg_info "$(translate 'Installing DKMS package...')" - dpkg -i ../gasket-dkms_*.deb || true + cd /tmp || exit 1 + rm -rf gasket-driver >>"$LOG_FILE" 2>&1 + msg_info "$(translate 'Cloning Google Coral driver repository...')" + git clone https://github.com/google/gasket-driver.git >>"$LOG_FILE" 2>&1 + if [[ $? -ne 0 ]]; then msg_error "$(translate 'Could not clone the repository. Check /tmp/coral_install.log')"; exit 1; fi + msg_ok "$(translate 'Repository cloned successfully.')" - # === Compile with DKMS === - msg_info "$(translate 'Compiling Coral TPU drivers for current kernel...')" - dkms add -m gasket -v 1.0 || true - dkms build -m gasket -v 1.0 -k "$(uname -r)" - dkms install -m gasket -v 1.0 -k "$(uname -r)" - # Load modules immediately - modprobe gasket 2>/dev/null - modprobe apex 2>/dev/null - if lsmod | grep -q apex; then - msg_success "$(translate 'Coral TPU drivers installed and loaded successfully.')" - else - msg_warn "$(translate 'Installation finished but drivers are not loaded. Please check dmesg.')" - fi + cd /tmp/gasket-driver || exit 1 + msg_info "$(translate 'Patching source for kernel compatibility...')" - echo -e + + sed -i 's/\.llseek = no_llseek/\.llseek = noop_llseek/' src/gasket_core.c + + sed -i 's/^MODULE_IMPORT_NS(DMA_BUF);/MODULE_IMPORT_NS("DMA_BUF");/' src/gasket_page_table.c + + sed -i "s/\(linux-headers-686-pae | linux-headers-amd64 | linux-headers-generic | linux-headers\)/\1 | proxmox-headers-$(uname -r) | pve-headers-$(uname -r)/" debian/control + if [[ $? -ne 0 ]]; then msg_error "$(translate 'Patching failed. Check /tmp/coral_install.log')"; exit 1; fi + msg_ok "$(translate 'Source patched successfully.')" + + + + msg_info "$(translate 'Building DKMS package...')" + debuild -us -uc -tc -b >>"$LOG_FILE" 2>&1 + if [[ $? -ne 0 ]]; then msg_error "$(translate 'Failed to build DKMS package. Check /tmp/coral_install.log')"; exit 1; fi + msg_ok "$(translate 'DKMS package built successfully.')" + + + + msg_info "$(translate 'Installing DKMS package...')" + dpkg -i ../gasket-dkms_*.deb >>"$LOG_FILE" 2>&1 || true + if ! dpkg -s gasket-dkms >/dev/null 2>&1; then + msg_error "$(translate 'Failed to install DKMS package. Check /tmp/coral_install.log')"; exit 1 + fi + msg_ok "$(translate 'DKMS package installed.')" + + + + msg_info "$(translate 'Compiling Coral TPU drivers for current kernel...')" + dkms remove -m gasket -v 1.0 -k "$(uname -r)" >>"$LOG_FILE" 2>&1 || true + dkms add -m gasket -v 1.0 >>"$LOG_FILE" 2>&1 || true + dkms build -m gasket -v 1.0 -k "$(uname -r)" >>"$LOG_FILE" 2>&1 + if [[ $? -ne 0 ]]; then + sed -n '1,200p' /var/lib/dkms/gasket/1.0/build/make.log >>"$LOG_FILE" 2>&1 || true + msg_error "$(translate 'DKMS build failed. Check /tmp/coral_install.log')"; exit 1 + fi + dkms install -m gasket -v 1.0 -k "$(uname -r)" >>"$LOG_FILE" 2>&1 + if [[ $? -ne 0 ]]; then msg_error "$(translate 'DKMS install failed. Check /tmp/coral_install.log')"; exit 1; fi + msg_ok "$(translate 'Drivers compiled and installed via DKMS.')" + + + + msg_info "$(translate 'Loading modules...')" + modprobe gasket >>"$LOG_FILE" 2>&1 || true + modprobe apex >>"$LOG_FILE" 2>&1 || true + if lsmod | grep -q '\bapex\b'; then + msg_ok "$(translate 'Modules loaded.')" + msg_success "$(translate 'Coral TPU drivers installed and loaded successfully.')" + else + msg_warn "$(translate 'Installation finished but drivers are not loaded. Please check dmesg and /tmp/coral_install.log')" + fi + + + + echo "---- dmesg | grep -i apex (last lines) ----" >>"$LOG_FILE" + dmesg | grep -i apex | tail -n 20 >>"$LOG_FILE" 2>&1 } -# Prompt for reboot after installation restart_prompt() { - if whiptail --title "$(translate 'Coral TPU Installation')" --yesno "$(translate 'The installation requires a server restart to apply changes. Do you want to restart now?')" 10 70; then - msg_warn "$(translate 'Restarting the server...')" - reboot - else - echo -e - msg_success "$(translate "Press Enter to return to menu...")" - read -r - fi + if whiptail --title "$(translate 'Coral TPU Installation')" --yesno \ + "$(translate 'The installation requires a server restart to apply changes. Do you want to restart now?')" 10 70; then + msg_warn "$(translate 'Restarting the server...')" + reboot + else + msg_success "$(translate 'Completed. Press Enter to return to menu...')" + read -r + fi } -# Main + pre_install_prompt install_coral_host restart_prompt