Fabian
56 lines
1.8 KiB
Bash
56 lines
1.8 KiB
Bash
#!/bin/bash
|
|
# =============================================================
|
|
# ComfyUI Launch Script for AMD BC-250 (ROCm / gfx1013 → gfx1030 spoof)
|
|
# =============================================================
|
|
set -euo pipefail
|
|
|
|
echo "=========================================="
|
|
echo " ComfyUI — BC-250 ROCm Launcher"
|
|
echo "=========================================="
|
|
|
|
# --- GPU Health Check ---
|
|
if dmesg 2>/dev/null | tail -50 | grep -qi "KIQ fence timeout"; then
|
|
echo "[ABORT] KIQ fence timeout detected in dmesg — reboot required!"
|
|
exit 1
|
|
fi
|
|
echo "[OK] GPU health check passed"
|
|
|
|
# --- ROCm Environment for BC-250 ---
|
|
# CRITICAL: gfx1030 spoof (not gfx1010!) — PyTorch ROCm 6.2 has no gfx1010 kernels
|
|
export HSA_OVERRIDE_GFX_VERSION=10.3.0
|
|
export HSA_ENABLE_SDMA=0
|
|
export HIP_VISIBLE_DEVICES=0
|
|
export ROCM_PATH=/opt/rocm
|
|
export HSA_TOOLS_LIB=""
|
|
export HSA_TOOLS_REPORT_LOAD_FAILURE=0
|
|
export PATH="/opt/rocm/bin:$PATH"
|
|
export LD_LIBRARY_PATH="/opt/rocm/lib"
|
|
|
|
# --- Performance: DO NOT set these ---
|
|
unset GPU_MAX_HW_QUEUES 2>/dev/null || true
|
|
unset HIP_LAUNCH_BLOCKING 2>/dev/null || true
|
|
unset GGML_CUDA_ENABLE_UNIFIED_MEMORY 2>/dev/null || true
|
|
unset GGML_HIP_HOST_ALLOC 2>/dev/null || true
|
|
unset GGML_CUDA_NO_PINNED 2>/dev/null || true
|
|
unset GGML_HIP_NO_COARSE_GRAIN 2>/dev/null || true
|
|
unset HSA_DISABLE_FRAGMENT_ALLOCATOR 2>/dev/null || true
|
|
|
|
# --- PyTorch ROCm tuning ---
|
|
export PYTORCH_HIP_ALLOC_CONF="expandable_segments:False"
|
|
|
|
echo "[OK] ROCm environment configured (gfx1030 spoof)"
|
|
|
|
# --- Activate venv ---
|
|
cd ~/ComfyUI
|
|
source venv/bin/activate
|
|
|
|
# --- Launch ComfyUI ---
|
|
echo "[START] Launching ComfyUI on http://0.0.0.0:8188"
|
|
echo "=========================================="
|
|
python main.py \
|
|
--listen 0.0.0.0 \
|
|
--port 8188 \
|
|
--force-fp32 \
|
|
--lowvram \
|
|
"$@"
|