mirror of
https://github.com/MacRimi/ProxMenux.git
synced 2025-06-28 04:06:54 +00:00
Update configure_coral.sh
This commit is contained in:
MacRimi
GitHub
parent
a1b0bcf3e0
commit
84ca427f10
@ -1,348 +1,3 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
# ==========================================================
|
|
||||||
# ProxMenu - A menu-driven script for Proxmox VE management
|
|
||||||
# ==========================================================
|
|
||||||
# Author : MacRimi
|
|
||||||
# Copyright : (c) 2024 MacRimi
|
|
||||||
# License : MIT (https://raw.githubusercontent.com/MacRimi/ProxMenux/main/LICENSE)
|
|
||||||
# Version : 1.0
|
|
||||||
# Last Updated: 28/01/2025
|
|
||||||
# ==========================================================
|
|
||||||
|
|
||||||
|
|
||||||
# Configuration ============================================
|
|
||||||
REPO_URL="https://raw.githubusercontent.com/MacRimi/ProxMenux/main"
|
|
||||||
UTILS_URL="https://raw.githubusercontent.com/MacRimi/ProxMenux/main/scripts/utils.sh"
|
|
||||||
BASE_DIR="/usr/local/share/proxmenux"
|
|
||||||
CACHE_FILE="$BASE_DIR/cache.json"
|
|
||||||
CONFIG_FILE="$BASE_DIR/config.json"
|
|
||||||
VENV_PATH="/opt/googletrans-env"
|
|
||||||
|
|
||||||
if ! source <(curl -sSf "$UTILS_URL"); then
|
|
||||||
echo "$(translate 'Error: Could not load utils.sh from') $UTILS_URL"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
initialize_cache() {
|
|
||||||
if [ ! -f "$CACHE_FILE" ]; then
|
|
||||||
echo "{}" > "$CACHE_FILE"
|
|
||||||
return
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
load_language() {
|
|
||||||
if [ -f "$CONFIG_FILE" ]; then
|
|
||||||
LANGUAGE=$(jq -r '.language' "$CONFIG_FILE")
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# ==========================================================
|
|
||||||
|
|
||||||
# Variables globales
|
|
||||||
CHANGES_MADE=0 # Inicializar cambios para verificar si es necesario reiniciar el contenedor
|
|
||||||
NEED_REBOOT=0 # Controla si se requiere reinicio completo del servidor
|
|
||||||
|
|
||||||
|
|
||||||
# Validar que el contenedor seleccionado es válido
|
|
||||||
validate_container_id() {
|
|
||||||
if [ -z "$CONTAINER_ID" ]; then
|
|
||||||
msg_error "ID del contenedor no definido. Asegúrate de seleccionar un contenedor primero."
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# Función para reinicio
|
|
||||||
restart_prompt() {
|
|
||||||
if (whiptail --title "Reinicio requerido" --yesno "La instalación requiere un reinicio del servidor para que los cambios sean efectivos. ¿Deseas reiniciar ahora?" 8 58); then
|
|
||||||
msg_info "Reiniciando el servidor..."
|
|
||||||
reboot
|
|
||||||
else
|
|
||||||
msg_info "No se realizó el reinicio. Por favor, reinicia manualmente más tarde para aplicar los cambios."
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# Reinicio del servidor al final
|
|
||||||
final_restart_prompt() {
|
|
||||||
if [ "$NEED_REBOOT" -eq 1 ]; then
|
|
||||||
msg_info "Se recomienda reiniciar el servidor para aplicar los cambios correctamente."
|
|
||||||
restart_prompt
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# Selección del contenedor LXC
|
|
||||||
select_container() {
|
|
||||||
CONTAINERS=$(pct list | awk 'NR>1 {print $1, $3}' | xargs -n2)
|
|
||||||
if [ -z "$CONTAINERS" ]; then
|
|
||||||
msg_error "No hay contenedores disponibles en Proxmox."
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
CONTAINER_ID=$(whiptail --title "Seleccionar Contenedor" \
|
|
||||||
--menu "Selecciona el contenedor LXC:" 15 60 5 $CONTAINERS 3>&1 1>&2 2>&3)
|
|
||||||
|
|
||||||
if [ -z "$CONTAINER_ID" ]; then
|
|
||||||
msg_error "No se seleccionó ningún contenedor. Saliendo."
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! pct list | awk 'NR>1 {print $1}' | grep -qw "$CONTAINER_ID"; then
|
|
||||||
msg_error "El contenedor con ID $CONTAINER_ID no existe. Saliendo."
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "Contenedor seleccionado: $CONTAINER_ID"
|
|
||||||
}
|
|
||||||
|
|
||||||
# Validar y cambiar a privilegiado si es necesario
|
|
||||||
ensure_privileged_container() {
|
|
||||||
validate_container_id
|
|
||||||
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
|
|
||||||
if [ ! -f "$CONFIG_FILE" ]; then
|
|
||||||
msg_error "Archivo de configuración del contenedor $CONTAINER_ID no encontrado."
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
if grep -q "^unprivileged: 1" "$CONFIG_FILE"; then
|
|
||||||
msg_info "El contenedor es no privilegiado. Cambiando a privilegiado..."
|
|
||||||
sed -i "s/^unprivileged: 1/unprivileged: 0/" "$CONFIG_FILE"
|
|
||||||
STORAGE_TYPE=$(pct config "$CONTAINER_ID" | grep "^rootfs:" | awk -F, '{print $2}' | cut -d'=' -f2)
|
|
||||||
if [[ "$STORAGE_TYPE" == "dir" ]]; then
|
|
||||||
STORAGE_PATH=$(pct config "$CONTAINER_ID" | grep "^rootfs:" | awk '{print $2}' | cut -d',' -f1)
|
|
||||||
chown -R root:root "$STORAGE_PATH"
|
|
||||||
fi
|
|
||||||
msg_ok "Contenedor cambiado a privilegiado."
|
|
||||||
else
|
|
||||||
msg_ok "El contenedor ya es privilegiado."
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# Verificar y configurar repositorios en el host
|
|
||||||
verify_and_add_repos() {
|
|
||||||
msg_info "Verificando y configurando repositorios necesarios en el host..."
|
|
||||||
|
|
||||||
if ! grep -q "pve-no-subscription" /etc/apt/sources.list /etc/apt/sources.list.d/* 2>/dev/null; then
|
|
||||||
echo "deb http://download.proxmox.com/debian/pve $(lsb_release -sc) pve-no-subscription" | tee /etc/apt/sources.list.d/pve-no-subscription.list
|
|
||||||
msg_ok "Repositorio pve-no-subscription añadido."
|
|
||||||
else
|
|
||||||
msg_ok "Repositorio pve-no-subscription ya configurado."
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! grep -q "non-free-firmware" /etc/apt/sources.list; then
|
|
||||||
echo "deb http://deb.debian.org/debian $(lsb_release -sc) main contrib non-free-firmware
|
|
||||||
deb http://deb.debian.org/debian $(lsb_release -sc)-updates main contrib non-free-firmware
|
|
||||||
deb http://security.debian.org/debian-security $(lsb_release -sc)-security main contrib non-free-firmware" | tee -a /etc/apt/sources.list
|
|
||||||
msg_ok "Repositorios non-free-firmware añadidos."
|
|
||||||
else
|
|
||||||
msg_ok "Repositorios non-free-firmware ya configurados."
|
|
||||||
fi
|
|
||||||
|
|
||||||
apt-get update
|
|
||||||
msg_ok "Repositorios verificados y actualizados."
|
|
||||||
}
|
|
||||||
|
|
||||||
# Configurar Coral TPU en el contenedor
|
|
||||||
configure_lxc_for_coral() {
|
|
||||||
ensure_privileged_container
|
|
||||||
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
|
|
||||||
|
|
||||||
# Verificar y agregar configuraciones solo si no existen
|
|
||||||
if ! grep -Pq "^lxc.cgroup2.devices.allow: c 189:\* rwm # Coral USB$" "$CONFIG_FILE"; then
|
|
||||||
echo "lxc.cgroup2.devices.allow: c 189:* rwm # Coral USB" >> "$CONFIG_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! grep -Pq "^lxc.mount.entry: /dev/bus/usb dev/bus/usb none bind,optional,create=dir$" "$CONFIG_FILE"; then
|
|
||||||
echo "lxc.mount.entry: /dev/bus/usb dev/bus/usb none bind,optional,create=dir" >> "$CONFIG_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! grep -Pq "^lxc.mount.entry: /dev/apex_0 dev/apex_0 none bind,optional,create=file$" "$CONFIG_FILE"; then
|
|
||||||
echo "lxc.mount.entry: /dev/apex_0 dev/apex_0 none bind,optional,create=file" >> "$CONFIG_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "Configuración de Coral TPU (USB y M.2) añadida al contenedor $CONTAINER_ID."
|
|
||||||
}
|
|
||||||
|
|
||||||
# Configurar iGPU en el contenedor
|
|
||||||
configure_lxc_for_igpu() {
|
|
||||||
ensure_privileged_container
|
|
||||||
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
|
|
||||||
|
|
||||||
# Verificar y agregar configuraciones solo si no existen
|
|
||||||
if ! grep -q "features: nesting=1" "$CONFIG_FILE"; then
|
|
||||||
echo "features: nesting=1" >> "$CONFIG_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! grep -q "c 226:0 rwm" "$CONFIG_FILE"; then
|
|
||||||
echo "lxc.cgroup2.devices.allow: c 226:0 rwm # iGPU" >> "$CONFIG_FILE"
|
|
||||||
echo "lxc.cgroup2.devices.allow: c 226:128 rwm # iGPU" >> "$CONFIG_FILE"
|
|
||||||
echo "lxc.mount.entry: /dev/dri dev/dri none bind,optional,create=dir" >> "$CONFIG_FILE"
|
|
||||||
echo "lxc.mount.entry: /dev/dri/renderD128 dev/dri/renderD128 none bind,optional,create=file" >> "$CONFIG_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! grep -q "c 29:0 rwm # Framebuffer" "$CONFIG_FILE"; then
|
|
||||||
echo "lxc.cgroup2.devices.allow: c 29:0 rwm # Framebuffer" >> "$CONFIG_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! grep -q "lxc.mount.entry: /dev/fb0" "$CONFIG_FILE"; then
|
|
||||||
echo "lxc.mount.entry: /dev/fb0 dev/fb0 none bind,optional,create=file" >> "$CONFIG_FILE"
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "Configuración de iGPU añadida al contenedor $CONTAINER_ID."
|
|
||||||
}
|
|
||||||
|
|
||||||
# Instalar controladores iGPU en el contenedor
|
|
||||||
install_igpu_in_container() {
|
|
||||||
msg_info "Instalando controladores de iGPU dentro del contenedor..."
|
|
||||||
pct start "$CONTAINER_ID"
|
|
||||||
pct exec "$CONTAINER_ID" -- bash -c "
|
|
||||||
apt-get update && \
|
|
||||||
apt-get install -y va-driver-all ocl-icd-libopencl1 intel-opencl-icd vainfo intel-gpu-tools && \
|
|
||||||
chgrp video /dev/dri && chmod 755 /dev/dri && \
|
|
||||||
adduser root video && adduser root render
|
|
||||||
"
|
|
||||||
msg_ok "Controladores de iGPU instalados dentro del contenedor."
|
|
||||||
}
|
|
||||||
|
|
||||||
# Instalar TPU en host y verificación.
|
|
||||||
NEED_REBOOT=0
|
|
||||||
install_coral_host() {
|
|
||||||
FORCE_REINSTALL=$1
|
|
||||||
|
|
||||||
# Verificar si el controlador ya está instalado
|
|
||||||
if [ "$FORCE_REINSTALL" != "--force" ]; then
|
|
||||||
msg_info "Verificando si los controladores de Coral TPU ya están instalados..."
|
|
||||||
if dpkg -l | grep -qw gasket-dkms; then
|
|
||||||
msg_ok "Los controladores de Coral TPU ya están instalados."
|
|
||||||
return 0
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_info "Instalando controladores de Coral TPU en el host..."
|
|
||||||
verify_and_add_repos
|
|
||||||
apt-get install -y git devscripts dh-dkms dkms pve-headers-$(uname -r)
|
|
||||||
|
|
||||||
# Clonar la rama predeterminada del repositorio
|
|
||||||
cd /tmp
|
|
||||||
rm -rf gasket-driver
|
|
||||||
msg_info "Clonando la rama predeterminada del repositorio de Google Coral..."
|
|
||||||
git clone https://github.com/google/gasket-driver.git
|
|
||||||
if [ $? -ne 0 ]; then
|
|
||||||
msg_error "No se pudo clonar el repositorio."
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
cd gasket-driver/
|
|
||||||
|
|
||||||
# Construir e instalar el paquete .deb
|
|
||||||
debuild -us -uc -tc -b
|
|
||||||
if [ $? -ne 0 ]; then
|
|
||||||
msg_error "Error al construir los paquetes del controlador."
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
dpkg -i ../gasket-dkms_*.deb
|
|
||||||
if [ $? -ne 0 ]; then
|
|
||||||
msg_error "Error al instalar los paquetes del controlador."
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "Controladores de Coral TPU instalados en el host desde la rama predeterminada."
|
|
||||||
NEED_REBOOT=1 # Marcar que se requiere reinicio completo del servidor
|
|
||||||
}
|
|
||||||
|
|
||||||
# Instalar controladores Coral TPU en el contenedor
|
|
||||||
install_coral_in_container() {
|
|
||||||
msg_info "Detectando dispositivos Coral TPU dentro del contenedor..."
|
|
||||||
CORAL_M2=$(lspci | grep -i "Global Unichip")
|
|
||||||
|
|
||||||
if [[ -n "$CORAL_M2" ]]; then
|
|
||||||
DRIVER_OPTION=$(whiptail --title "Seleccionar versión de controladores" \
|
|
||||||
--menu "Elige la versión de controladores para Coral M.2:\n\nPrecaución: El modo máximo genera más calor." 15 60 2 \
|
|
||||||
1 "libedgetpu1-std (rendimiento estándar)" \
|
|
||||||
2 "libedgetpu1-max (máximo rendimiento)" 3>&1 1>&2 2>&3)
|
|
||||||
|
|
||||||
case "$DRIVER_OPTION" in
|
|
||||||
1) DRIVER_PACKAGE="libedgetpu1-std" ;;
|
|
||||||
2) DRIVER_PACKAGE="libedgetpu1-max" ;;
|
|
||||||
*) DRIVER_PACKAGE="libedgetpu1-std" ;;
|
|
||||||
esac
|
|
||||||
else
|
|
||||||
DRIVER_PACKAGE="libedgetpu1-std"
|
|
||||||
fi
|
|
||||||
|
|
||||||
pct start "$CONTAINER_ID"
|
|
||||||
pct exec "$CONTAINER_ID" -- bash -c "
|
|
||||||
apt-get update && \
|
|
||||||
apt-get install -y gnupg python3 python3-pip python3-venv && \
|
|
||||||
curl -fsSL https://packages.cloud.google.com/apt/doc/apt-key.gpg | gpg --dearmor -o /usr/share/keyrings/coral-edgetpu.gpg && \
|
|
||||||
echo 'deb [signed-by=/usr/share/keyrings/coral-edgetpu.gpg] https://packages.cloud.google.com/apt coral-edgetpu-stable main' | tee /etc/apt/sources.list.d/coral-edgetpu.list && \
|
|
||||||
apt-get update && \
|
|
||||||
apt-get install -y $DRIVER_PACKAGE
|
|
||||||
"
|
|
||||||
msg_ok "Controladores de Coral TPU instalados dentro del contenedor."
|
|
||||||
}
|
|
||||||
|
|
||||||
# Logica del del proceso de instalar coral
|
|
||||||
if dpkg -l | grep -qw gasket-dkms; then
|
|
||||||
msg_info "Controladores de Coral ya están instalados en el host."
|
|
||||||
|
|
||||||
if (whiptail --title "Reinstalar controladores" --yesno "¿Quieres reinstalar los controladores de Coral en el servidor?\n(Esto solo es necesario en caso de error)" 10 60); then
|
|
||||||
msg_info "Reinstalando los controladores de Coral en el host..."
|
|
||||||
install_coral_host --force
|
|
||||||
configure_lxc_for_coral && CHANGES_MADE=1
|
|
||||||
install_coral_in_container && CHANGES_MADE=1
|
|
||||||
configure_lxc_for_igpu && CHANGES_MADE=1
|
|
||||||
install_igpu_in_container && CHANGES_MADE=1
|
|
||||||
msg_info "Reiniciando el servicio udev para aplicar cambios..."
|
|
||||||
systemctl restart udev
|
|
||||||
msg_ok "El servicio udev ha sido reiniciado con éxito. No se requiere reinicio completo del servidor."
|
|
||||||
else
|
|
||||||
configure_lxc_for_coral && CHANGES_MADE=1
|
|
||||||
install_coral_in_container && CHANGES_MADE=1
|
|
||||||
configure_lxc_for_igpu && CHANGES_MADE=1
|
|
||||||
install_igpu_in_container && CHANGES_MADE=1
|
|
||||||
if [ "$CHANGES_MADE" -eq 1 ]; then
|
|
||||||
if pct status "$CONTAINER_ID" | grep -q "running"; then
|
|
||||||
pct restart "$CONTAINER_ID"
|
|
||||||
else
|
|
||||||
pct start "$CONTAINER_ID"
|
|
||||||
fi
|
|
||||||
msg_ok "El contenedor $CONTAINER_ID ha sido reiniciado para aplicar los cambios de Coral TPU."
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
msg_info "Instalando controladores de Coral por primera vez en el host..."
|
|
||||||
install_coral_host
|
|
||||||
configure_lxc_for_coral && CHANGES_MADE=1
|
|
||||||
install_coral_in_container && CHANGES_MADE=1
|
|
||||||
configure_lxc_for_igpu && CHANGES_MADE=1
|
|
||||||
install_igpu_in_container && CHANGES_MADE=1
|
|
||||||
if pct status "$CONTAINER_ID" | grep -q "running"; then
|
|
||||||
pct restart "$CONTAINER_ID"
|
|
||||||
else
|
|
||||||
pct start "$CONTAINER_ID"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "Configuración completada."
|
|
||||||
|
|
||||||
# Solicitar reinicio al final si es necesario
|
|
||||||
final_restart_prompt
|
|
||||||
|
|
||||||
# Iniciar
|
|
||||||
validate_pve_version
|
|
||||||
select_container
|
|
||||||
final_restart_prompt
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
# ==========================================================
|
# ==========================================================
|
||||||
@ -393,40 +48,9 @@ load_language() {
|
|||||||
LANGUAGE=$(jq -r '.language' "$CONFIG_FILE")
|
LANGUAGE=$(jq -r '.language' "$CONFIG_FILE")
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# ==========================================================
|
# ==========================================================
|
||||||
|
|
||||||
# Global variables
|
# Select LXC container
|
||||||
CHANGES_MADE=0 # Initialize changes to check if container restart is necessary
|
|
||||||
NEED_REBOOT=0 # Controls if a full server reboot is required
|
|
||||||
|
|
||||||
# Validate that the selected container is valid
|
|
||||||
validate_container_id() {
|
|
||||||
if [ -z "$CONTAINER_ID" ]; then
|
|
||||||
msg_error "$(translate 'Container ID not defined. Make sure to select a container first.')"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# Function for restart
|
|
||||||
restart_prompt() {
|
|
||||||
if (whiptail --title "$(translate 'Restart Required')" --yesno "$(translate 'The installation requires a server restart for changes to take effect. Do you want to restart now?')" 8 58); then
|
|
||||||
msg_info "$(translate 'Restarting the server...')"
|
|
||||||
reboot
|
|
||||||
else
|
|
||||||
msg_info "$(translate 'Restart not performed. Please restart manually later to apply changes.')"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# Server restart at the end
|
|
||||||
final_restart_prompt() {
|
|
||||||
if [ "$NEED_REBOOT" -eq 1 ]; then
|
|
||||||
msg_info "$(translate 'It is recommended to restart the server to apply changes correctly.')"
|
|
||||||
restart_prompt
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
# LXC container selection
|
|
||||||
select_container() {
|
select_container() {
|
||||||
CONTAINERS=$(pct list | awk 'NR>1 {print $1, $3}' | xargs -n2)
|
CONTAINERS=$(pct list | awk 'NR>1 {print $1, $3}' | xargs -n2)
|
||||||
if [ -z "$CONTAINERS" ]; then
|
if [ -z "$CONTAINERS" ]; then
|
||||||
@ -450,8 +74,18 @@ select_container() {
|
|||||||
msg_ok "$(translate 'Container selected:') $CONTAINER_ID"
|
msg_ok "$(translate 'Container selected:') $CONTAINER_ID"
|
||||||
}
|
}
|
||||||
|
|
||||||
# Validate and change to privileged if necessary
|
|
||||||
ensure_privileged_container() {
|
# Validate that the selected container is valid
|
||||||
|
validate_container_id() {
|
||||||
|
if [ -z "$CONTAINER_ID" ]; then
|
||||||
|
msg_error "$(translate 'Container ID not defined. Make sure to select a container first.')"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
# Configure LXC for Coral TPU and iGPU
|
||||||
|
configure_lxc_hardware() {
|
||||||
validate_container_id
|
validate_container_id
|
||||||
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
|
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
|
||||||
if [ ! -f "$CONFIG_FILE" ]; then
|
if [ ! -f "$CONFIG_FILE" ]; then
|
||||||
@ -471,38 +105,8 @@ ensure_privileged_container() {
|
|||||||
else
|
else
|
||||||
msg_ok "$(translate 'The container is already privileged.')"
|
msg_ok "$(translate 'The container is already privileged.')"
|
||||||
fi
|
fi
|
||||||
}
|
|
||||||
|
|
||||||
# Verify and configure repositories on the host
|
# Configure Coral TPU (USB and M.2)
|
||||||
verify_and_add_repos() {
|
|
||||||
msg_info "$(translate 'Verifying and configuring necessary repositories on the host...')"
|
|
||||||
|
|
||||||
if ! grep -q "pve-no-subscription" /etc/apt/sources.list /etc/apt/sources.list.d/* 2>/dev/null; then
|
|
||||||
echo "deb http://download.proxmox.com/debian/pve $(lsb_release -sc) pve-no-subscription" | tee /etc/apt/sources.list.d/pve-no-subscription.list
|
|
||||||
msg_ok "$(translate 'pve-no-subscription repository added.')"
|
|
||||||
else
|
|
||||||
msg_ok "$(translate 'pve-no-subscription repository already configured.')"
|
|
||||||
fi
|
|
||||||
|
|
||||||
if ! grep -q "non-free-firmware" /etc/apt/sources.list; then
|
|
||||||
echo "deb http://deb.debian.org/debian $(lsb_release -sc) main contrib non-free-firmware
|
|
||||||
deb http://deb.debian.org/debian $(lsb_release -sc)-updates main contrib non-free-firmware
|
|
||||||
deb http://security.debian.org/debian-security $(lsb_release -sc)-security main contrib non-free-firmware" | tee -a /etc/apt/sources.list
|
|
||||||
msg_ok "$(translate 'non-free-firmware repositories added.')"
|
|
||||||
else
|
|
||||||
msg_ok "$(translate 'non-free-firmware repositories already configured.')"
|
|
||||||
fi
|
|
||||||
|
|
||||||
apt-get update
|
|
||||||
msg_ok "$(translate 'Repositories verified and updated.')"
|
|
||||||
}
|
|
||||||
|
|
||||||
# Configure Coral TPU in the container
|
|
||||||
configure_lxc_for_coral() {
|
|
||||||
ensure_privileged_container
|
|
||||||
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
|
|
||||||
|
|
||||||
# Verify and add configurations only if they don't exist
|
|
||||||
if ! grep -Pq "^lxc.cgroup2.devices.allow: c 189:\* rwm # Coral USB$" "$CONFIG_FILE"; then
|
if ! grep -Pq "^lxc.cgroup2.devices.allow: c 189:\* rwm # Coral USB$" "$CONFIG_FILE"; then
|
||||||
echo "lxc.cgroup2.devices.allow: c 189:* rwm # Coral USB" >> "$CONFIG_FILE"
|
echo "lxc.cgroup2.devices.allow: c 189:* rwm # Coral USB" >> "$CONFIG_FILE"
|
||||||
fi
|
fi
|
||||||
@ -515,15 +119,7 @@ configure_lxc_for_coral() {
|
|||||||
echo "lxc.mount.entry: /dev/apex_0 dev/apex_0 none bind,optional,create=file" >> "$CONFIG_FILE"
|
echo "lxc.mount.entry: /dev/apex_0 dev/apex_0 none bind,optional,create=file" >> "$CONFIG_FILE"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
msg_ok "$(translate 'Coral TPU configuration (USB and M.2) added to container') $CONTAINER_ID."
|
# Configure iGPU
|
||||||
}
|
|
||||||
|
|
||||||
# Configure iGPU in the container
|
|
||||||
configure_lxc_for_igpu() {
|
|
||||||
ensure_privileged_container
|
|
||||||
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
|
|
||||||
|
|
||||||
# Verify and add configurations only if they don't exist
|
|
||||||
if ! grep -q "features: nesting=1" "$CONFIG_FILE"; then
|
if ! grep -q "features: nesting=1" "$CONFIG_FILE"; then
|
||||||
echo "features: nesting=1" >> "$CONFIG_FILE"
|
echo "features: nesting=1" >> "$CONFIG_FILE"
|
||||||
fi
|
fi
|
||||||
@ -543,74 +139,17 @@ configure_lxc_for_igpu() {
|
|||||||
echo "lxc.mount.entry: /dev/fb0 dev/fb0 none bind,optional,create=file" >> "$CONFIG_FILE"
|
echo "lxc.mount.entry: /dev/fb0 dev/fb0 none bind,optional,create=file" >> "$CONFIG_FILE"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
msg_ok "$(translate 'iGPU configuration added to container') $CONTAINER_ID."
|
msg_ok "$(translate 'Coral TPU and iGPU configuration added to container') $CONTAINER_ID."
|
||||||
}
|
}
|
||||||
|
|
||||||
# Install iGPU drivers in the container
|
|
||||||
install_igpu_in_container() {
|
|
||||||
msg_info "$(translate 'Installing iGPU drivers inside the container...')"
|
|
||||||
pct start "$CONTAINER_ID"
|
|
||||||
pct exec "$CONTAINER_ID" -- bash -c "
|
|
||||||
apt-get update && \
|
|
||||||
apt-get install -y va-driver-all ocl-icd-libopencl1 intel-opencl-icd vainfo intel-gpu-tools && \
|
|
||||||
chgrp video /dev/dri && chmod 755 /dev/dri && \
|
|
||||||
adduser root video && adduser root render
|
|
||||||
"
|
|
||||||
msg_ok "$(translate 'iGPU drivers installed inside the container.')"
|
|
||||||
}
|
|
||||||
|
|
||||||
# Install TPU on host and verification.
|
|
||||||
NEED_REBOOT=0
|
|
||||||
install_coral_host() {
|
|
||||||
FORCE_REINSTALL=$1
|
|
||||||
|
|
||||||
# Check if the driver is already installed
|
|
||||||
if [ "$FORCE_REINSTALL" != "--force" ]; then
|
|
||||||
msg_info "$(translate 'Checking if Coral TPU drivers are already installed...')"
|
|
||||||
if dpkg -l | grep -qw gasket-dkms; then
|
|
||||||
msg_ok "$(translate 'Coral TPU drivers are already installed.')"
|
|
||||||
return 0
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_info "$(translate 'Installing Coral TPU drivers on the host...')"
|
|
||||||
verify_and_add_repos
|
|
||||||
apt-get install -y git devscripts dh-dkms dkms pve-headers-$(uname -r)
|
|
||||||
|
|
||||||
# Clone the default branch of the repository
|
|
||||||
cd /tmp
|
|
||||||
rm -rf gasket-driver
|
|
||||||
msg_info "$(translate 'Cloning the default branch of the Google Coral repository...')"
|
|
||||||
git clone https://github.com/google/gasket-driver.git
|
|
||||||
if [ $? -ne 0 ]; then
|
|
||||||
msg_error "$(translate 'Could not clone the repository.')"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
cd gasket-driver/
|
|
||||||
|
|
||||||
# Build and install the .deb package
|
|
||||||
debuild -us -uc -tc -b
|
|
||||||
if [ $? -ne 0 ]; then
|
|
||||||
msg_error "$(translate 'Error building the driver packages.')"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
dpkg -i ../gasket-dkms_*.deb
|
|
||||||
if [ $? -ne 0 ]; then
|
|
||||||
msg_error "$(translate 'Error installing the driver packages.')"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "$(translate 'Coral TPU drivers installed on the host from the default branch.')"
|
|
||||||
NEED_REBOOT=1 # Mark that a full server reboot is required
|
|
||||||
}
|
|
||||||
|
|
||||||
# Install Coral TPU drivers in the container
|
# Install Coral TPU drivers in the container
|
||||||
install_coral_in_container() {
|
install_coral_in_container() {
|
||||||
msg_info "$(translate 'Detecting Coral TPU devices inside the container...')"
|
echo -ne "\r${TAB}${YW}- $(translate 'Installing iGPU and Coral TPU drivers inside the container...') ${CL}"
|
||||||
CORAL_M2=$(lspci | grep -i "Global Unichip")
|
|
||||||
|
|
||||||
|
pct start "$CONTAINER_ID"
|
||||||
|
|
||||||
|
CORAL_M2=$(lspci | grep -i "Global Unichip")
|
||||||
if [[ -n "$CORAL_M2" ]]; then
|
if [[ -n "$CORAL_M2" ]]; then
|
||||||
DRIVER_OPTION=$(whiptail --title "$(translate 'Select driver version')" \
|
DRIVER_OPTION=$(whiptail --title "$(translate 'Select driver version')" \
|
||||||
--menu "$(translate 'Choose the driver version for Coral M.2:\n\nCaution: Maximum mode generates more heat.')" 15 60 2 \
|
--menu "$(translate 'Choose the driver version for Coral M.2:\n\nCaution: Maximum mode generates more heat.')" 15 60 2 \
|
||||||
@ -626,66 +165,40 @@ install_coral_in_container() {
|
|||||||
DRIVER_PACKAGE="libedgetpu1-std"
|
DRIVER_PACKAGE="libedgetpu1-std"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
pct start "$CONTAINER_ID"
|
pct exec "$CONTAINER_ID" -- bash <<EOF
|
||||||
pct exec "$CONTAINER_ID" -- bash -c "
|
set -e
|
||||||
apt-get update && \
|
|
||||||
apt-get install -y gnupg python3 python3-pip python3-venv && \
|
echo "- Updating package lists..."
|
||||||
curl -fsSL https://packages.cloud.google.com/apt/doc/apt-key.gpg | gpg --dearmor -o /usr/share/keyrings/coral-edgetpu.gpg && \
|
apt-get update
|
||||||
echo 'deb [signed-by=/usr/share/keyrings/coral-edgetpu.gpg] https://packages.cloud.google.com/apt coral-edgetpu-stable main' | tee /etc/apt/sources.list.d/coral-edgetpu.list && \
|
|
||||||
apt-get update && \
|
echo "- Installing iGPU drivers..."
|
||||||
apt-get install -y $DRIVER_PACKAGE
|
apt-get install -y va-driver-all ocl-icd-libopencl1 intel-opencl-icd vainfo intel-gpu-tools
|
||||||
"
|
chgrp video /dev/dri && chmod 755 /dev/dri
|
||||||
msg_ok "$(translate 'Coral TPU drivers installed inside the container.')"
|
adduser root video && adduser root render
|
||||||
|
|
||||||
|
echo "- Installing Coral TPU dependencies..."
|
||||||
|
apt-get install -y gnupg python3 python3-pip python3-venv
|
||||||
|
|
||||||
|
echo "- Adding Coral TPU repository..."
|
||||||
|
curl -fsSL https://packages.cloud.google.com/apt/doc/apt-key.gpg | gpg --dearmor -o /usr/share/keyrings/coral-edgetpu.gpg
|
||||||
|
echo 'deb [signed-by=/usr/share/keyrings/coral-edgetpu.gpg] https://packages.cloud.google.com/apt coral-edgetpu-stable main' | tee /etc/apt/sources.list.d/coral-edgetpu.list
|
||||||
|
|
||||||
|
echo "- Updating package lists again..."
|
||||||
|
apt-get update
|
||||||
|
|
||||||
|
echo "- Installing Coral TPU driver (\$DRIVER_PACKAGE)..."
|
||||||
|
apt-get install -y \$DRIVER_PACKAGE
|
||||||
|
EOF
|
||||||
|
|
||||||
|
|
||||||
|
msg_ok "$(translate 'iGPU and Coral TPU drivers installed inside the container.')"
|
||||||
}
|
}
|
||||||
|
|
||||||
# Logic of the Coral installation process
|
|
||||||
if dpkg -l | grep -qw gasket-dkms; then
|
|
||||||
msg_info "$(translate 'Coral drivers are already installed on the host.')"
|
|
||||||
|
|
||||||
if (whiptail --title "$(translate 'Reinstall drivers')" --yesno "$(translate 'Do you want to reinstall Coral drivers on the server?\n(This is only necessary in case of error)')" 10 60); then
|
select_container
|
||||||
msg_info "$(translate 'Reinstalling Coral drivers on the host...')"
|
configure_lxc_hardware
|
||||||
install_coral_host --force
|
install_coral_in_container
|
||||||
configure_lxc_for_coral && CHANGES_MADE=1
|
|
||||||
install_coral_in_container && CHANGES_MADE=1
|
|
||||||
configure_lxc_for_igpu && CHANGES_MADE=1
|
|
||||||
install_igpu_in_container && CHANGES_MADE=1
|
|
||||||
msg_info "$(translate 'Restarting the udev service to apply changes...')"
|
|
||||||
systemctl restart udev
|
|
||||||
msg_ok "$(translate 'The udev service has been successfully restarted. No full server reboot required.')"
|
|
||||||
else
|
|
||||||
configure_lxc_for_coral && CHANGES_MADE=1
|
|
||||||
install_coral_in_container && CHANGES_MADE=1
|
|
||||||
configure_lxc_for_igpu && CHANGES_MADE=1
|
|
||||||
install_igpu_in_container && CHANGES_MADE=1
|
|
||||||
if [ "$CHANGES_MADE" -eq 1 ]; then
|
|
||||||
if pct status "$CONTAINER_ID" | grep -q "running"; then
|
|
||||||
pct restart "$CONTAINER_ID"
|
|
||||||
else
|
|
||||||
pct start "$CONTAINER_ID"
|
|
||||||
fi
|
|
||||||
msg_ok "$(translate 'Container') $CONTAINER_ID $(translate 'has been restarted to apply Coral TPU changes.')"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
msg_info "$(translate 'Installing Coral drivers for the first time on the host...')"
|
|
||||||
install_coral_host
|
|
||||||
configure_lxc_for_coral && CHANGES_MADE=1
|
|
||||||
install_coral_in_container && CHANGES_MADE=1
|
|
||||||
configure_lxc_for_igpu && CHANGES_MADE=1
|
|
||||||
install_igpu_in_container && CHANGES_MADE=1
|
|
||||||
if pct status "$CONTAINER_ID" | grep -q "running"; then
|
|
||||||
pct restart "$CONTAINER_ID"
|
|
||||||
else
|
|
||||||
pct start "$CONTAINER_ID"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
msg_ok "$(translate 'Configuration completed.')"
|
msg_ok "$(translate 'Configuration completed.')"
|
||||||
|
|
||||||
# Request restart at the end if necessary
|
|
||||||
final_restart_prompt
|
|
||||||
|
|
||||||
# Start
|
|
||||||
validate_pve_version
|
|
||||||
select_container
|
|
||||||
final_restart_prompt
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user