Update add_hw_acceleration_lxc.sh

This commit is contained in:
MacRimi 2024-12-20 19:37:16 +01:00 committed by GitHub
parent f7ac842e82
commit b2db51d7ee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,12 +1,12 @@
#!/bin/bash
# Funciones y colores
# Colores para salida
YW="\033[33m"; GN="\033[1;92m"; RD="\033[01;31m"; CL="\033[m"
msg_info() { echo -ne " ${YW}[INFO] $1...${CL}"; }
msg_ok() { echo -e " ${GN}[OK] $1${CL}"; }
msg_error() { echo -e " ${RD}[ERROR] $1${CL}"; }
# Validaciones
# Validar la versión de Proxmox
validate_pve_version() {
if ! pveversion | grep -Eq "pve-manager/(8\\.[0-9]+)"; then
msg_error "Se requiere Proxmox VE 8.0 o superior."
@ -14,6 +14,7 @@ validate_pve_version() {
fi
}
# Selección del contenedor LXC
select_container() {
CONTAINERS=$(pct list | awk 'NR>1 {print $1, $3}' | xargs -n2)
CONTAINER_ID=$(whiptail --title "Seleccionar Contenedor" --menu "Selecciona un contenedor LXC:" 15 60 5 $CONTAINERS 3>&1 1>&2 2>&3)
@ -25,10 +26,52 @@ select_container() {
msg_ok "Contenedor seleccionado: $CONTAINER_ID"
}
# Asegurarse de que el contenedor sea privilegiado
ensure_privileged_container() {
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
# Validar que el archivo de configuración exista
if [ ! -f "$CONFIG_FILE" ]; then
msg_error "Archivo de configuración del contenedor $CONTAINER_ID no encontrado."
exit 1
fi
# Verificar si el contenedor es no privilegiado
if grep -q "^unprivileged: 1" "$CONFIG_FILE"; then
msg_info "El contenedor es no privilegiado. Cambiando a privilegiado..."
# Cambiar la configuración a privilegiado
sed -i "s/^unprivileged: 1/unprivileged: 0/" "$CONFIG_FILE"
# Ajustar permisos del sistema de archivos si el almacenamiento es 'dir'
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"
msg_ok "Permisos ajustados para almacenamiento en directorios."
fi
msg_ok "Contenedor cambiado a privilegiado."
else
msg_ok "El contenedor ya es privilegiado."
fi
}
# Apagar el contenedor si está corriendo
stop_container_if_running() {
if pct status "$CONTAINER_ID" | grep -q "status: running"; then
msg_info "Apagando el contenedor $CONTAINER_ID..."
pct stop "$CONTAINER_ID"
msg_ok "Contenedor $CONTAINER_ID apagado."
fi
}
# Configurar iGPU en el contenedor
configure_lxc_for_igpu() {
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
CONFIG_PRESENT=true
# Verificar cada línea específica para la configuración de iGPU
if ! grep -q "lxc.cgroup2.devices.allow: c 226:0 rwm" "$CONFIG_FILE"; then
CONFIG_PRESENT=false
fi
@ -66,9 +109,16 @@ configure_lxc_for_igpu() {
msg_ok "Configuración de iGPU añadida al contenedor."
}
install_igpu_in_container() {
# Iniciar el contenedor
start_container() {
msg_info "Iniciando el contenedor $CONTAINER_ID..."
pct start "$CONTAINER_ID"
msg_ok "Contenedor $CONTAINER_ID iniciado."
}
# Instalar controladores de iGPU en el contenedor
install_igpu_in_container() {
msg_info "Instalando controladores de iGPU dentro del contenedor..."
pct exec "$CONTAINER_ID" -- bash -c "
apt-get update && \
apt-get install -y va-driver-all intel-opencl-icd vainfo && \
@ -80,6 +130,9 @@ install_igpu_in_container() {
# Ejecución
validate_pve_version
select_container
ensure_privileged_container
stop_container_if_running
configure_lxc_for_igpu
start_container
install_igpu_in_container
msg_ok "Aceleración gráfica iGPU configurada correctamente."