Esegui Stable Diffusion con ComfyUI su un server GPU

Eseguire la generazione di immagini sul tuo server GPU significa nessun log dei prompt in mano a terzi, nessun filtro sui contenuti che non hai scelto e un costo che resta fisso, per quante immagini tu faccia. Questa guida configura ComfyUI, un'interfaccia a nodi per Stable Diffusion e modelli simili, su Ubuntu.

Scegli la GPU

La memoria video è il vincolo. 8 GB fanno girare SD 1.5 senza problemi, 12-16 GB gestiscono bene SDXL e 24 GB lasciano spazio a modelli più recenti e grandi e a risoluzioni più alte. Se hai dubbi, vedi GPU consumer contro GPU da datacenter. Lascia anche circa 50 GB di disco, perché i file dei modelli sono grandi.

Passo 1: controlla il driver

nvidia-smi

Se mostra la tua GPU, il driver funziona. Altrimenti installa il driver consigliato con sudo ubuntu-drivers autoinstall e riavvia.

Passo 2: installa 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

Fai corrispondere l'indice delle wheel CUDA alla versione che PyTorch consiglia per il tuo driver. Controlla la loro pagina di installazione se il comando qui sopra è cambiato.

Passo 3: aggiungi un modello

Scarica un checkpoint che hai la licenza di usare in ComfyUI/models/checkpoints/. Leggi la licenza del modello per l'uso commerciale prima di costruirci sopra un prodotto.

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

Passo 4: avvialo in privato

Per impostazione predefinita ComfyUI ascolta solo su localhost, ed è quello che vuoi. Raggiungilo tramite un tunnel SSH.

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

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

Apri http://127.0.0.1:8188 sul tuo computer. Non esporre la porta a internet: per impostazione predefinita non c'è alcun login.

Passo 5: mantienilo in esecuzione

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

Adatta utente e percorsi al tuo server.

Costi e privacy a confronto con gli strumenti ospitati

• Costo: una GPU a ore costa uguale che tu faccia dieci immagini o diecimila. Conviene rispetto al prezzo per immagine quando generi in grandi quantità e non conviene quando generi di rado, quindi ferma il server quando è inattivo.
• Privacy: prompt e risultati restano su una macchina che controlli tu e puoi pagare con Monero o altre monete senza documento d'identità.
• Controllo: scegli tu il modello, il sampler e le impostazioni di sicurezza, e ti assumi la responsabilità di come li usi.

Per i modelli di testo sullo stesso hardware, vedi Ollama e Open WebUI.