# Verifique os recursos do sistema
df -h          # Espaço em disco
free -h        # Memória RAM
nvidia-smi     # GPU (se instalada)

# Atualize o sistema e instale as ferramentas
sudo apt update
sudo apt install -y ca-certificates curl tar

# Adicione a chave GPG oficial do 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

# Adicione o repositório 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 o Docker
sudo apt update
sudo apt install -y docker-ce docker-ce-cli containerd.io docker-buildx-plugin docker-compose-plugin

# Adicione o usuário ao grupo docker (opcional)
sudo usermod -aG docker $USER

# Verifique a instalação
docker --version
docker compose version

# Remova drivers antigos e instale novos
sudo apt purge 'nvidia-*'
sudo apt autoremove
sudo apt install -y nvidia-driver-580

# Reinicie o servidor
sudo reboot

# Após reiniciar, verifique a GPU
nvidia-smi

# Adicione o repositório e instale o 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 o Docker para GPU
sudo nvidia-ctk runtime configure --runtime=docker
sudo systemctl restart docker

# Crie um diretório para os pacotes
mkdir -p nvidia-offline-packages/drivers
cd nvidia-offline-packages/drivers

# Baixe o driver via apt download (se o repositório estiver disponível)
# Primeiro adicione o repositório da NVIDIA (se ainda não estiver adicionado)
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update

# Baixe o pacote do driver
apt download nvidia-driver-580

# Baixe as dependências do driver (execute o comando linha por linha)
# Primeiro descubra a lista de dependências:
apt-cache depends nvidia-driver-580 | grep "Depends:" | cut -d: -f2 | tr -d ' ' > /tmp/deps.txt

# Em seguida, baixe cada dependência:
while read dep; do
	apt download "$dep" 2>/dev/null || echo "Dependência ignorada: $dep"
done < /tmp/deps.txt

# Exclua o arquivo temporário
rm /tmp/deps.txt

# Volte para o diretório dos pacotes
cd nvidia-offline-packages

# Crie um diretório para o Container Toolkit
mkdir -p toolkit

# Adicione o repositório do NVIDIA Container Toolkit (se ainda não estiver adicionado)
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

# Baixe o pacote do Container Toolkit e todas as suas dependências
cd toolkit
apt download nvidia-container-toolkit

# Baixe as dependências do Container Toolkit (execute o comando linha por linha)
# Primeiro descubra a lista de dependências:
apt-cache depends nvidia-container-toolkit | grep "Depends:" | cut -d: -f2 | tr -d ' ' > /tmp/toolkit-deps.txt

# Em seguida, baixe cada dependência:
while read dep; do
  apt download "$dep" 2>/dev/null || echo "Dependência ignorada: $dep"
done < /tmp/toolkit-deps.txt

# Exclua o arquivo temporário
rm /tmp/toolkit-deps.txt

# Opcional: Baixe a chave GPG e o arquivo do repositório (necessários apenas se você planeja usar apt install em vez de dpkg)
# Para instalação offline via dpkg -i, as chaves GPG NÃO são necessárias
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

# Volte para o diretório raiz dos pacotes
cd nvidia-offline-packages

# Verifique se todos os arquivos estão no lugar:
# - drivers/ - contém pacotes .deb ou arquivo .run do driver
# - toolkit/ - contém pacotes .deb do Container Toolkit
# - keys/ - contém chaves GPG e arquivos do repositório (opcional, se houver)

# Crie um arquivo com todos os pacotes
# Se o diretório keys/ estiver vazio ou ausente, você pode criar o arquivo sem ele:
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 "Atenção: Chaves GPG não incluídas no arquivo (não são obrigatórias para instalação via dpkg)"
fi

# Verifique o tamanho do arquivo
ls -lh nvidia-packages-offline.tar.gz

# Extraia o arquivo
tar -xzf nvidia-packages-offline.tar.gz
cd nvidia-offline-packages

# Verifique a presença de arquivos .deb no diretório drivers
ls -la drivers/*.deb

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

# Verifique o resultado da instalação
if [ $? -eq 0 ]; then
    echo "Drivers instalados com sucesso"
else
    echo "Erros de dependência encontrados, corrigindo..."
    # Se houver erros de dependência, instale-os do cache ou corrija as dependências
    sudo apt install --fix-broken -y
    
    # Se o fix-broken não ajudou, tente instalar as dependências manualmente
    # Verifique se todas as dependências foram baixadas para o diretório drivers
    echo "Verifique se todas as dependências foram baixadas para o diretório drivers/"
    ls -la *.deb
fi

# Reinicie o servidor para aplicar os drivers
echo "Reiniciando o servidor em 10 segundos... Pressione Ctrl+C para cancelar"
sleep 10
sudo reboot

**Se a instalação falhar:**

* Verifique os logs: `sudo cat /var/log/nvidia-installer.log`
* Certifique-se de que o módulo nouveau está desativado: `lsmod | grep nouveau`
* Tente a instalação com a flag `--no-opengl-files` se o erro estiver relacionado ao OpenGL

<details>

<summary>💡 Comentários sobre a instalação do arquivo .run</summary>

**Flags do instalador:**

* `--silent` - instalação automática sem perguntas interativas
* `--no-nouveau-check` - ignora a verificação do módulo nouveau (driver aberto da NVIDIA)
* `--no-opengl-files` - não instala bibliotecas OpenGL (importante para servidores sem GUI)
* `--disable-nouveau` - desativa o módulo nouveau antes da instalação

**Parando o servidor gráfico:**

* Em servidores sem interface gráfica, esta etapa não é necessária
* Se o servidor gráfico não for parado, a instalação pode falhar
* Após a instalação, o servidor gráfico pode ser iniciado novamente

**Método alternativo (sem parar o GUI):** Se parar o servidor gráfico não for possível, você pode usar o console de texto:

1. Mude para o console de texto (Ctrl+Alt+F1)
2. Faça login no sistema
3. Execute a instalação do arquivo .run
4. Volte para o modo gráfico (Ctrl+Alt+F7 ou reinicie)

</details>

**Verificação da instalação dos drivers**

Após reiniciar, verifique a instalação dos drivers:

```bash
# Verifique o GPU (este comando deve mostrar informações sobre a placa de vídeo)
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 |
| ...                                                                         |
+-----------------------------------------------------------------------------+

# Volte para o diretório dos pacotes (substitua pelo caminho real onde você descompactou o arquivo)
# Por exemplo, se o arquivo foi descompactado no diretório home:
cd ~/nvidia-offline-packages
# Ou se em outro lugar, especifique o caminho completo:
# cd /caminho/completo/para/nvidia-offline-packages

# Verifique se todos os arquivos estão no lugar
ls -la toolkit/*.deb

# Opcional: Instale a chave GPG (NÃO é obrigatório para instalação via dpkg)
# As chaves GPG são necessárias apenas se:
# 1. Você planeja usar apt install em vez de dpkg -i
# 2. Você deseja configurar o repositório para futuras atualizações (quando a internet estiver disponível)
# Para uma instalação offline simples via dpkg -i, você pode pular esta etapa
if [ -f "keys/nvidia-container-toolkit.gpg" ]; then
    echo "Instalando chave 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 "Chave GPG instalada"
else
    echo "Chave GPG não encontrada - pulando instalação (não crítico para dpkg -i)"
fi

# Instale o Container Toolkit a partir dos pacotes locais
cd toolkit
sudo dpkg -i *.deb

# Verifique o resultado da instalação
if [ $? -eq 0 ]; then
    echo "Container Toolkit instalado com sucesso"
else
    echo "Erros de dependência encontrados, corrigindo..."
    # Corrija as dependências se necessário
    sudo apt install --fix-broken -y
    
    # Se o fix-broken não ajudou, verifique se todas as dependências foram baixadas
    echo "Verifique a presença de todas as dependências no diretório toolkit/"
    ls -la *.deb
fi

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

# Reinicie o Docker para aplicar as alterações
sudo systemctl restart docker

# Verifique se o Docker está em execução
sudo systemctl status docker

# Verifique a instalação do GPU no Docker
nvidia-smi