Execute o Stable Diffusion com ComfyUI em um servidor GPU

Rodar geração de imagens no seu próprio servidor GPU significa nenhum log de prompts em poder de terceiros, nenhum filtro de conteúdo que você não escolheu e um custo que permanece fixo, não importa quantas imagens você faça. Este guia configura o ComfyUI, uma interface baseada em nós para o Stable Diffusion e modelos semelhantes, no Ubuntu.

Escolha a GPU

A memória de vídeo é a limitação. 8 GB rodam o SD 1.5 com folga, 12 a 16 GB lidam bem com o SDXL e 24 GB dão espaço para modelos novos e maiores e resoluções mais altas. Se estiver em dúvida, veja GPUs de consumo versus de datacenter. Deixe também cerca de 50 GB de disco, porque os arquivos de modelo são grandes.

Passo 1: verifique o driver

nvidia-smi

Se ele mostrar a sua GPU, o driver funciona. Se não, instale o driver recomendado com sudo ubuntu-drivers autoinstall e reinicie.

Passo 2: instale o ComfyUI

sudo apt update && sudo apt install -y git python3-venv
git clone https://github.com/comfyanonymous/ComfyUI
cd ComfyUI
python3 -m venv venv
. venv/bin/activate
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements.txt

Faça o índice de wheels CUDA corresponder à versão que o PyTorch recomenda para o seu driver. Consulte a página de instalação deles se o comando acima tiver mudado.

Passo 3: adicione um modelo

Baixe um checkpoint que você tenha licença para usar em ComfyUI/models/checkpoints/. Leia a licença do modelo quanto ao uso comercial antes de construir um produto em cima dele.

cd models/checkpoints
wget -O model.safetensors "https://example.com/path/to/model.safetensors"

Passo 4: inicie de forma privada

Por padrão, o ComfyUI escuta apenas em localhost, que é o que você quer. Acesse-o por um túnel SSH.

python main.py --listen 127.0.0.1 --port 8188

ssh -L 8188:127.0.0.1:8188 user@your-server

Abra http://127.0.0.1:8188 no seu próprio computador. Não exponha a porta à internet: não há login por padrão.

Passo 5: mantenha-o em execução

sudo tee /etc/systemd/system/comfyui.service > /dev/null <<'UNIT'
[Unit]
Description=ComfyUI
After=network-online.target

[Service]
User=ubuntu
WorkingDirectory=/home/ubuntu/ComfyUI
ExecStart=/home/ubuntu/ComfyUI/venv/bin/python main.py --listen 127.0.0.1
Restart=on-failure

[Install]
WantedBy=multi-user.target
UNIT
sudo systemctl daemon-reload && sudo systemctl enable --now comfyui

Ajuste o usuário e os caminhos para corresponder ao seu servidor.

Custo e privacidade em comparação com ferramentas hospedadas

• Custo: uma GPU por hora custa o mesmo se você fizer dez imagens ou dez mil. Ela vence o preço por imagem quando você gera em volume e perde quando gera raramente, então pare o servidor quando ocioso.
• Privacidade: prompts e resultados ficam em uma máquina que você controla, e você pode pagar com Monero ou outras moedas sem documento de identidade.
• Controle: você escolhe o modelo, o sampler e as configurações de segurança, e assume a responsabilidade pelo modo como os usa.

Para modelos de texto no mesmo hardware, veja Ollama e Open WebUI.