# Verifique los recursos del sistema
df -h          # Espacio en disco
free -h        # Memoria RAM
nvidia-smi     # GPU (si está instalada)

# Actualice el sistema e instale herramientas
sudo apt update
sudo apt install -y ca-certificates curl tar

# Agregue la clave GPG oficial de Docker
sudo install -m 0755 -d /etc/apt/keyrings
sudo curl -fsSL https://download.docker.com/linux/ubuntu/gpg -o /etc/apt/keyrings/docker.asc
sudo chmod a+r /etc/apt/keyrings/docker.asc

# Agregue el repositorio de Docker
echo "deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.asc] https://download.docker.com/linux/ubuntu \
  $(. /etc/os-release && echo "${UBUNTU_CODENAME:-$VERSION_CODENAME}") stable" | \
  sudo tee /etc/apt/sources.list.d/docker.list > /dev/null

# Instale Docker
sudo apt update
sudo apt install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin

# Agregue el usuario al grupo docker (opcional)
sudo usermod -aG docker $USER

# Verifique la instalación
docker --version
docker compose version

# Elimine los controladores antiguos e instale los nuevos
sudo apt purge 'nvidia-*'
sudo apt autoremove
sudo apt install -y nvidia-driver-580

# Reinicie el servidor
sudo reboot

# Después de reiniciar, verifique la GPU
nvidia-smi

# Agregue el repositorio e instale el toolkit
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | \
  sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit.gpg

curl -fsSL https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
  sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit.gpg] https://#g' | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

sudo apt update
sudo apt install -y nvidia-container-toolkit

# Configure Docker para GPU
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker

# Cree un directorio para los paquetes
mkdir -p nvidia-offline-packages/drivers
cd nvidia-offline-packages/drivers

# Descargue el controlador a través de apt download (si el repositorio está disponible)
# Primero, agregue el repositorio de NVIDIA (si aún no se ha agregado)
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

# Descargue el paquete del controlador
apt download nvidia-driver-580

# Descargue las dependencias del controlador (ejecute el comando línea por línea)
# Primero, obtenga la lista de dependencias:
apt-cache depends nvidia-driver-580 | grep "Depends:" | cut -d: -f2 | tr -d ' ' > /tmp/deps.txt

# Luego, descargue cada dependencia:
while read dep; do
	apt download "$dep" 2>/dev/null || echo "Dependencia omitida: $dep"
done < /tmp/deps.txt

# Elimine el archivo temporal
rm /tmp/deps.txt

# Regrese al directorio de paquetes
cd nvidia-offline-packages

# Cree un directorio para el Container Toolkit
mkdir -p toolkit

# Agregue el repositorio del NVIDIA Container Toolkit (si aún no se ha agregado)
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | \
  sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit.gpg

curl -fsSL https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
  sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit.gpg] https://#g' | \
  sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list

sudo apt update

# Descargue el paquete del Container Toolkit y todas sus dependencias
cd toolkit
apt download nvidia-container-toolkit

# Descargue las dependencias del Container Toolkit (ejecute el comando línea por línea)
# Primero, obtenga la lista de dependencias:
apt-cache depends nvidia-container-toolkit | grep "Depends:" | cut -d: -f2 | tr -d ' ' > /tmp/toolkit-deps.txt

# Luego, descargue cada dependencia:
while read dep; do
  apt download "$dep" 2>/dev/null || echo "Dependencia omitida: $dep"
done < /tmp/toolkit-deps.txt

# Elimine el archivo temporal
rm /tmp/toolkit-deps.txt

# Opcional: Descargue la clave GPG y el archivo del repositorio (solo son necesarios si planea usar apt install en lugar de dpkg)
# Para la instalación sin conexión a través de dpkg -i, las claves GPG NO son necesarias
cd ..
mkdir -p keys
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey -o keys/nvidia-container-toolkit.gpg
curl -fsSL https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list -o keys/nvidia-container-toolkit.list

# Regrese al directorio raíz de los paquetes
cd nvidia-offline-packages

# Asegúrese de que todos los archivos estén en su lugar:
# - drivers/ - contiene paquetes .deb o el archivo .run del controlador
# - toolkit/ - contiene paquetes .deb del Container Toolkit
# - keys/ - contiene claves GPG y archivos del repositorio (opcional, si existen)

# Cree un archivo comprimido con todos los paquetes
# Si el directorio keys/ está vacío o no existe, se puede crear el archivo comprimido sin él:
if [ -d "keys" ] && [ "$(ls -A keys)" ]; then
    tar -czf nvidia-packages-offline.tar.gz drivers/ toolkit/ keys/
else
    tar -czf nvidia-packages-offline.tar.gz drivers/ toolkit/
    echo "Advertencia: Las claves GPG no están incluidas en el archivo (no son obligatorias para la instalación a través de dpkg)"
fi

# Verifique el tamaño del archivo comprimido
ls -lh nvidia-packages-offline.tar.gz

# Descomprima el archivo
tar -xzf nvidia-packages-offline.tar.gz
cd nvidia-offline-packages

# Verifique la presencia de archivos .deb en el directorio drivers
ls -la drivers/*.deb

# Instale los controladores NVIDIA
cd drivers
sudo dpkg -i *.deb

# Verifique el resultado de la instalación
if [ $? -eq 0 ]; then
    echo "Controladores instalados exitosamente"
else
    echo "Se encontraron errores de dependencias, corrigiendo..."
    # Si hubo errores de dependencias, instálelas desde la caché o corrija las dependencias
    sudo apt install --fix-broken -y
    
    # Si fix-broken no ayudó, intente instalar las dependencias manualmente
    # Asegúrese de que todas las dependencias se hayan descargado en el directorio drivers
    echo "Verifique que todas las dependencias se hayan descargado en el directorio drivers/"
    ls -la *.deb
fi

# Reinicie el servidor para aplicar los controladores
echo "Reiniciando el servidor en 10 segundos... Presione Ctrl+C para cancelar"
sleep 10
sudo reboot

**Si la instalación terminó con un error:**

* Verifique los registros: `sudo cat /var/log/nvidia-installer.log`
* Asegúrese de que el módulo nouveau esté deshabilitado: `lsmod | grep nouveau`
* Intente la instalación con el flag `--no-opengl-files` si el error está relacionado con OpenGL

<details>

<summary>💡 Comentarios sobre la instalación del archivo .run</summary>

**Flags del instalador:**

* `--silent` - instalación automática sin preguntas interactivas
* `--no-nouveau-check` - omite la verificación del módulo nouveau (controlador abierto de NVIDIA)
* `--no-opengl-files` - no instala bibliotecas de OpenGL (importante para servidores sin GUI)
* `--disable-nouveau` - desactiva el módulo nouveau antes de la instalación

**Detención del servidor gráfico:**

* En servidores sin interfaz gráfica, este paso no es necesario
* Si el servidor gráfico no está detenido, la instalación puede fallar
* Después de la instalación, el servidor gráfico se puede reiniciar

**Método alternativo (sin detener la GUI):** Si no es posible detener el servidor gráfico, se puede utilizar la consola de texto:

1. Cambie a la consola de texto (Ctrl+Alt+F1)
2. Inicie sesión
3. Realice la instalación del archivo .run
4. Regrese al modo gráfico (Ctrl+Alt+F7 o reinicie)

</details>

**Verificación de la instalación de los controladores**

Después de reiniciar, verifique la instalación de los controladores:

```bash
# Verifique la GPU (este comando debe mostrar información sobre la tarjeta gráfica)
nvidia-smi

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 580.XX       Driver Version: 580.XX       CUDA Version: 12.X  |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00000000:01:00.0 Off |                  N/A |
| ...                                                                         |
+-----------------------------------------------------------------------------+

# Regrese al directorio de paquetes (reemplace con la ruta real donde descomprimió el archivo)
# Por ejemplo, si el archivo se descomprimió en el directorio personal:
cd ~/nvidia-offline-packages
# O si está en otro lugar, indique la ruta completa:
# cd /ruta/completa/a/nvidia-offline-packages

# Verifique que todos los archivos estén en su lugar
ls -la toolkit/*.deb

# Opcional: Instale la clave GPG (NO es obligatorio para la instalación a través de dpkg)
# Las claves GPG son necesarias solo si:
# 1. Planea usar apt install en lugar de dpkg -i
# 2. Desea configurar el repositorio para futuras actualizaciones (cuando haya internet)
# Para una instalación simple fuera de línea a través de dpkg -i, se puede omitir este paso
if [ -f "keys/nvidia-container-toolkit.gpg" ]; then
    echo "Instalando clave GPG (opcional)..."
    sudo mkdir -p /usr/share/keyrings
    sudo cp keys/nvidia-container-toolkit.gpg /tmp/nvidia-container-toolkit.gpg
    sudo gpg --dearmor /tmp/nvidia-container-toolkit.gpg -o /usr/share/keyrings/nvidia-container-toolkit.gpg
    sudo rm /tmp/nvidia-container-toolkit.gpg
    echo "Clave GPG instalada"
else
    echo "Clave GPG no encontrada - omitiendo instalación (no crítico para dpkg -i)"
fi

# Instale el Container Toolkit desde los paquetes locales
cd toolkit
sudo dpkg -i *.deb

# Verifique el resultado de la instalación
if [ $? -eq 0 ]; then
    echo "Container Toolkit instalado exitosamente"
else
    echo "Se encontraron errores de dependencias, corrigiendo..."
    # Corrija las dependencias si es necesario
    sudo apt install --fix-broken -y
    
    # Si fix-broken no ayudó, verifique que todas las dependencias estén descargadas
    echo "Verifique la presencia de todas las dependencias en el directorio toolkit/"
    ls -la *.deb
fi

# Configure Docker para GPU
sudo nvidia-ctk runtime configure --runtime=docker

# Reinicie Docker para aplicar los cambios
sudo systemctl restart docker

# Verifique que Docker esté en ejecución
sudo systemctl status docker

# Verifique la instalación de la GPU en Docker
nvidia-smi