fix: use --no-deps for AMD Windows torch wheel install

uv's resolver looks up rocm[libraries]==0.1.dev0 on PyPI during
dependency resolution before downloading any wheels, and fails because
the package doesn't exist on PyPI. --no-deps skips resolution entirely
and installs all 5 AMD wheels directly. The GPU runtime dependency is
satisfied by the HIP SDK, not a Python package.
This commit is contained in:
LeoBorcherding 2026-05-06 18:48:57 -05:00
commit 79670ab4a9
2 changed files with 2 additions and 1 deletions

View file

@ -334,6 +334,7 @@ def _ensure_rocm_torch() -> None:
f"ROCm torch (Windows, {rel_tag})",
"--force-reinstall",
"--no-cache-dir",
"--no-deps",
*wheel_urls,
constrain = False,
)