mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2025-10-01 23:56:18 +00:00
Update install_coral_pve9.sh
This commit is contained in:
@@ -1,103 +1,128 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
# ProxMenux - Coral TPU Installer (PVE 9.x)
|
# ProxMenux - Coral TPU Installer (PVE 9.x)
|
||||||
# =========================================
|
# =========================================
|
||||||
# Author : MacRimi
|
# Author : MacRimi
|
||||||
# License : MIT
|
# License : MIT
|
||||||
# Version : 1.1 (PVE9)
|
# Version : 1.3 (PVE9, silent build)
|
||||||
# Last Updated: 25/09/2025
|
# Last Updated: 25/09/2025
|
||||||
# =========================================
|
# =========================================
|
||||||
|
|
||||||
REPO_URL="https://raw.githubusercontent.com/MacRimi/ProxMenux/main"
|
REPO_URL="https://raw.githubusercontent.com/MacRimi/ProxMenux/main"
|
||||||
BASE_DIR="/usr/local/share/proxmenux"
|
BASE_DIR="/usr/local/share/proxmenux"
|
||||||
UTILS_FILE="$BASE_DIR/utils.sh"
|
UTILS_FILE="$BASE_DIR/utils.sh"
|
||||||
|
LOG_FILE="/tmp/coral_install.log"
|
||||||
|
|
||||||
if [[ -f "$UTILS_FILE" ]]; then
|
if [[ -f "$UTILS_FILE" ]]; then
|
||||||
source "$UTILS_FILE"
|
source "$UTILS_FILE"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
|
||||||
load_language
|
load_language
|
||||||
initialize_cache
|
initialize_cache
|
||||||
|
|
||||||
# Prompt before installation
|
|
||||||
pre_install_prompt() {
|
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
|
if ! dialog --title "$(translate 'Coral TPU Installation')" --yesno \
|
||||||
msg_warn "$(translate 'Installation cancelled by user.')"
|
"\n$(translate 'Installing Coral TPU drivers requires rebooting the server after installation. Do you want to proceed?')" 10 70; then
|
||||||
exit 0
|
|
||||||
fi
|
exit 0
|
||||||
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# Function to install Coral TPU drivers on PVE 9
|
|
||||||
install_coral_host() {
|
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 ===
|
cd /tmp || exit 1
|
||||||
msg_info "$(translate 'Installing DKMS package...')"
|
rm -rf gasket-driver >>"$LOG_FILE" 2>&1
|
||||||
dpkg -i ../gasket-dkms_*.deb || true
|
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
|
cd /tmp/gasket-driver || exit 1
|
||||||
msg_success "$(translate 'Coral TPU drivers installed and loaded successfully.')"
|
msg_info "$(translate 'Patching source for kernel compatibility...')"
|
||||||
else
|
|
||||||
msg_warn "$(translate 'Installation finished but drivers are not loaded. Please check dmesg.')"
|
|
||||||
fi
|
|
||||||
|
|
||||||
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() {
|
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
|
if whiptail --title "$(translate 'Coral TPU Installation')" --yesno \
|
||||||
msg_warn "$(translate 'Restarting the server...')"
|
"$(translate 'The installation requires a server restart to apply changes. Do you want to restart now?')" 10 70; then
|
||||||
reboot
|
msg_warn "$(translate 'Restarting the server...')"
|
||||||
else
|
reboot
|
||||||
echo -e
|
else
|
||||||
msg_success "$(translate "Press Enter to return to menu...")"
|
msg_success "$(translate 'Completed. Press Enter to return to menu...')"
|
||||||
read -r
|
read -r
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# Main
|
|
||||||
pre_install_prompt
|
pre_install_prompt
|
||||||
install_coral_host
|
install_coral_host
|
||||||
restart_prompt
|
restart_prompt
|
||||||
|
Reference in New Issue
Block a user