mirror of
https://github.com/community-scripts/ProxmoxVE.git
synced 2026-02-07 22:03:26 +01:00
Compare commits
4 Commits
openprojec
...
comfyui-cu
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
f1c8c7b7b4 | ||
|
|
a2a435041a | ||
|
|
87324564c6 | ||
|
|
c8dc7f2196 |
@@ -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"
|
||||
|
||||
Reference in New Issue
Block a user