ProxMenux/scripts/configure_igpu.sh

89 lines
2.5 KiB
Bash
Raw Normal View History

2025-01-30 11:35:06 +01:00
#!/bin/bash
# Variables globales
CHANGES_MADE=0 # Indica si se hicieron cambios en el contenedor
# Colores para salida
YW="\033[33m"
GN="\033[1;92m"
RD="\033[01;31m"
CL="\033[m"
# Funciones auxiliares
log() {
echo -e "$(date '+%Y-%m-%d %H:%M:%S') - $1"
}
msg_info() {
echo -ne " ${YW}[INFO] $1...${CL}"
}
msg_ok() {
echo -e " ${GN}[OK] $1${CL}"
}
msg_error() {
echo -e " ${RD}[ERROR] $1${CL}"
}
# 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
}
# 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
msg_ok "Contenedor seleccionado: $CONTAINER_ID"
}
# Configurar iGPU en el contenedor
configure_lxc_for_igpu() {
validate_container_id
CONFIG_FILE="/etc/pve/lxc/${CONTAINER_ID}.conf"
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"
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."
}
# Ejecución
select_container
configure_lxc_for_igpu
install_igpu_in_container
msg_ok "Configuración de iGPU completada en el contenedor $CONTAINER_ID."