Skip to content

Use CPU-only PyTorch wheels in uv lock - #1251

Merged
DingmaomaoBJTU merged 2 commits into
mainfrom
dingmaomaobjtu-adjust-uv-lock
Jul 29, 2026
Merged

Use CPU-only PyTorch wheels in uv lock#1251
DingmaomaoBJTU merged 2 commits into
mainfrom
dingmaomaobjtu-adjust-uv-lock

Conversation

@DingmaomaoBJTU

@DingmaomaoBJTU DingmaomaoBJTU commented Jul 29, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • Configure uv to resolve torch and torchvision from the CPU-only PyTorch index only for the Linux x86_64 resolution environment used by docs setup.
  • Keep Windows AMD64 torch and torchvision on the default PyPI wheels.
  • Regenerate uv.lock to remove CUDA/NVIDIA/Triton packages while preserving reproducible Linux docs locking.
  • Add a lockfile guard test to prevent CUDA/NVIDIA packages from returning.

Validation

  • uv lock --locked
  • uv run --no-project --with ruff ruff check tests\unit\test_uv_lock.py
  • uv run --no-project --with pytest --with pytest-cov --with pytest-timeout python -m pytest --confcutdir=tests\unit tests\unit\test_uv_lock.py -q
  • rg -i "cuda|cudnn|nvidia|cublas|cufft|curand|cusolver|cusparse|nccl|nvjitlink|nvtx|triton" uv.lock returns no matches

@DingmaomaoBJTU
DingmaomaoBJTU requested a review from a team as a code owner July 29, 2026 08:20
@DingmaomaoBJTU
DingmaomaoBJTU merged commit 45ca658 into main Jul 29, 2026
9 checks passed
@DingmaomaoBJTU
DingmaomaoBJTU deleted the dingmaomaobjtu-adjust-uv-lock branch July 29, 2026 08:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants