Compare commits

..

4 Commits

Author SHA1 Message Date
MickLesk
f1c8c7b7b4 Optimize PyTorch URL extraction: only parse for selected GPU type
Instead of extracting all three GPU URLs unconditionally, now only
extracts the URL for the user's selected GPU type. More efficient:
- Only one grep execution instead of three
- Less code duplication
- Faster execution
2026-02-07 18:05:14 +01:00
MickLesk
a2a435041a Simplify PyTorch version extraction: use local README instead of curl
Use the already-downloaded README.md from /opt/ComfyUI instead of
fetching it separately via curl. Much simpler and more reliable.

Benefits:
- No extra network request needed
- Guaranteed to match the installed ComfyUI version
- Fewer dependencies (no curl check)
- Faster execution
2026-02-07 18:03:59 +01:00
MickLesk
87324564c6 Make ComfyUI PyTorch versions dynamic from upstream README
Instead of hardcoding CUDA/ROCm/XPU versions, now dynamically extracts
the recommended PyTorch wheel URLs from ComfyUI's README.md at install time.

Benefits:
- Automatically stays up-to-date with upstream recommendations
- No manual updates needed when ComfyUI changes PyTorch versions
- Fallback to known-good versions (cu130, rocm6.4, xpu) if fetch fails

Implementation:
- Fetches https://raw.githubusercontent.com/comfyanonymous/ComfyUI/master/README.md
- Parses stable (non-nightly) pip install commands with regex
- Gracefully degrades to hardcoded fallbacks on network/parse errors
2026-02-07 18:03:16 +01:00
MickLesk
c8dc7f2196 Update ComfyUI PyTorch versions to match upstream recommendations
- NVIDIA: cu128 → cu130 (requires driver ≥ 570)
- AMD: rocm6.3 → rocm6.4 (stable version)
- Intel: xpu (unchanged, already matches upstream)

Fixes NVML version mismatch issues when host driver is upgraded
Aligns with ComfyUI upstream: https://github.com/Comfy-Org/ComfyUI#manual-install-windows-linux
2026-02-07 18:00:08 +01:00
7 changed files with 25 additions and 9 deletions

View File

@@ -11,7 +11,7 @@ var_cpu="${var_cpu:-2}"
var_ram="${var_ram:-2048}"
var_disk="${var_disk:-8}"
var_os="${var_os:-debian}"
var_version="${var_version:-12}"
var_version="${var_version:-13}"
var_unprivileged="${var_unprivileged:-1}"
header_info "$APP"

View File

@@ -11,7 +11,7 @@ var_cpu="${var_cpu:-2}"
var_ram="${var_ram:-4096}"
var_disk="${var_disk:-8}"
var_os="${var_os:-debian}"
var_version="${var_version:-12}"
var_version="${var_version:-13}"
var_unprivileged="${var_unprivileged:-1}"
header_info "$APP"

View File

@@ -11,7 +11,7 @@ var_cpu="${var_cpu:-2}"
var_ram="${var_ram:-2048}"
var_disk="${var_disk:-4}"
var_os="${var_os:-debian}"
var_version="${var_version:-12}"
var_version="${var_version:-13}"
var_unprivileged="${var_unprivileged:-1}"
header_info "$APP"

View File

@@ -25,7 +25,7 @@
"ram": 2048,
"hdd": 8,
"os": "debian",
"version": "12"
"version": "13"
}
}
],

View File

@@ -23,7 +23,7 @@
"ram": 4096,
"hdd": 8,
"os": "Debian",
"version": "12"
"version": "13"
}
}
],

View File

@@ -23,7 +23,7 @@
"ram": 2048,
"hdd": 4,
"os": "debian",
"version": "12"
"version": "13"
}
}
],

View File

@@ -37,26 +37,42 @@ fetch_and_deploy_gh_release "ComfyUI" "comfyanonymous/ComfyUI" "tarball" "latest
msg_info "Python dependencies"
$STD uv venv "/opt/ComfyUI/venv"
if [[ "${comfyui_gpu_type,,}" == "nvidia" ]]; then
pytorch_url="https://download.pytorch.org/whl/cu130"
if [[ -f "/opt/ComfyUI/README.md" ]]; then
extracted=$(grep -oP 'pip install.*?--extra-index-url\s+\Khttps://download\.pytorch\.org/whl/cu\d+' /opt/ComfyUI/README.md | head -1 || true)
[[ -n "$extracted" ]] && pytorch_url="$extracted"
fi
$STD uv pip install \
torch \
torchvision \
torchaudio \
--extra-index-url "https://download.pytorch.org/whl/cu128" \
--extra-index-url "$pytorch_url" \
--python="/opt/ComfyUI/venv/bin/python"
elif [[ "${comfyui_gpu_type,,}" == "amd" ]]; then
pytorch_url="https://download.pytorch.org/whl/rocm6.4"
if [[ -f "/opt/ComfyUI/README.md" ]]; then
extracted=$(grep -oP 'pip install.*?--index-url\s+\Khttps://download\.pytorch\.org/whl/rocm[\d.]+' /opt/ComfyUI/README.md | grep -v 'nightly' | head -1 || true)
[[ -n "$extracted" ]] && pytorch_url="$extracted"
fi
$STD uv pip install \
torch \
torchvision \
torchaudio \
--index-url "https://download.pytorch.org/whl/rocm6.3" \
--index-url "$pytorch_url" \
--python="/opt/ComfyUI/venv/bin/python"
elif [[ "${comfyui_gpu_type,,}" == "intel" ]]; then
pytorch_url="https://download.pytorch.org/whl/xpu"
if [[ -f "/opt/ComfyUI/README.md" ]]; then
extracted=$(grep -oP 'pip install.*?--index-url\s+\Khttps://download\.pytorch\.org/whl/xpu' /opt/ComfyUI/README.md | head -1 || true)
[[ -n "$extracted" ]] && pytorch_url="$extracted"
fi
$STD uv pip install \
torch \
torchvision \
torchaudio \
--index-url "https://download.pytorch.org/whl/xpu" \
--index-url "$pytorch_url" \
--python="/opt/ComfyUI/venv/bin/python"
fi
$STD uv pip install -r "/opt/ComfyUI/requirements.txt" --python="/opt/ComfyUI/venv/bin/python"