Chạy Stable Diffusion với ComfyUI trên máy chủ GPU

Chạy tạo ảnh trên máy chủ GPU của riêng bạn nghĩa là không có bên thứ ba lưu nhật ký prompt, không có bộ lọc nội dung mà bạn không chọn và chi phí cố định dù bạn tạo bao nhiêu ảnh. Hướng dẫn này cài đặt ComfyUI, một giao diện dựa trên node cho Stable Diffusion và các mô hình tương tự, trên Ubuntu.

Chọn GPU

Bộ nhớ video là yếu tố giới hạn. 8 GB chạy SD 1.5 thoải mái, 12 đến 16 GB xử lý SDXL tốt và 24 GB cho phép chạy các mô hình mới, lớn hơn và độ phân giải cao hơn. Nếu chưa chắc, hãy xem GPU phổ thông so với GPU trung tâm dữ liệu. Ngoài ra hãy chừa khoảng 50 GB ổ đĩa vì các tệp mô hình rất lớn.

Bước 1: kiểm tra driver

nvidia-smi

Nếu nó in ra GPU của bạn thì driver đang hoạt động. Nếu không, hãy cài driver được khuyến nghị bằng sudo ubuntu-drivers autoinstall rồi khởi động lại.

Bước 2: cài đặt 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

Hãy khớp chỉ mục wheel CUDA với phiên bản mà PyTorch khuyến nghị cho driver của bạn. Kiểm tra trang cài đặt của họ nếu lệnh ở trên đã lỗi thời.

Bước 3: thêm một mô hình

Tải một checkpoint mà bạn được cấp phép sử dụng vào ComfyUI/models/checkpoints/. Hãy đọc giấy phép của mô hình về việc dùng thương mại trước khi xây dựng sản phẩm trên đó.

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

Bước 4: khởi động ở chế độ riêng tư

Theo mặc định ComfyUI chỉ lắng nghe trên localhost, đúng như bạn muốn. Hãy truy cập nó qua đường hầm SSH.

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

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

Mở http://127.0.0.1:8188 trên máy tính của bạn. Đừng mở cổng này ra internet: theo mặc định không có đăng nhập.

Bước 5: giữ cho nó luôn chạy

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

Hãy điều chỉnh người dùng và đường dẫn cho khớp với máy chủ của bạn.

So sánh chi phí và quyền riêng tư với các công cụ lưu trữ sẵn

• Chi phí: GPU tính theo giờ có giá như nhau dù bạn tạo mười ảnh hay mười nghìn ảnh. Nó rẻ hơn giá theo từng ảnh khi bạn tạo hàng loạt, và đắt hơn khi bạn hiếm khi tạo, vì vậy hãy dừng máy chủ khi nhàn rỗi.
• Quyền riêng tư: prompt và kết quả đầu ra nằm trên máy bạn kiểm soát, và bạn có thể thanh toán bằng Monero hoặc các đồng tiền khác mà không cần giấy tờ tùy thân.
• Quyền kiểm soát: bạn chọn mô hình, bộ lấy mẫu và các thiết lập an toàn, đồng thời tự chịu trách nhiệm về cách sử dụng chúng.

Với các mô hình văn bản trên cùng phần cứng, hãy xem Ollama và Open WebUI.