Skip to content

build: install the torch build that matches the machine #1446

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
onuralpszr merged 1 commit into from
Aug 22, 2026
Merged

build: install the torch build that matches the machine #1446

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions .github/workflows/ci.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,8 @@ jobs:
HF_HOME: ${{ github.workspace }}/.cache/huggingface
# Give flaky HF Hub downloads more headroom before timing out (default is 10s)
HF_HUB_DOWNLOAD_TIMEOUT: "60"
# The runners have no GPU, so take the CPU wheels rather than detecting
UV_TORCH_BACKEND: cpu
strategy:
fail-fast: false
matrix:
Expand Down
19 changes: 9 additions & 10 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -105,8 +105,8 @@ all = ["sahi[yolov5,ultralytics,torchvision,transformers,roboflow,onnx]"]

ci = [
"sahi[transformers,roboflow,onnx]",
# pin the floors the torch extra declares; a bare version also matches the
# +cpu local build that the pytorch-cpu index serves Linux and Windows
# pin the floors the torch extra declares. A bare version also matches the
# local build tag, so these hold whichever backend uv resolves.
"torch==2.13.0;python_version>='3.10'",
"torch==2.8.0;python_version=='3.9'",
"torch==2.4.1;python_version=='3.8'",
Expand All @@ -121,6 +121,13 @@ ci = [

[tool.uv]

# A torch wheel bundles its own CUDA runtime, so the wheel has to match the
# machine rather than the requirement. "auto" reads the installed driver and
# picks the matching build, falling back to the CPU wheels where there is no
# GPU. CI pins it to cpu through UV_TORCH_BACKEND so the runners never pull a
# CUDA wheel they cannot use.
torch-backend = "auto"

# Keep every OpenCV distribution on the same version: they all install into the
# same `cv2/` directory, so mixing majors (e.g. opencv-python 5.x with a
# transitively pinned opencv-python-headless 4.x) breaks `import cv2`.
Expand All @@ -133,14 +140,6 @@ override-dependencies = [
"pywinpty!=2.0.14; os_name == 'nt'",
]

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true

[tool.uv.sources]
torch = [{ index = "pytorch-cpu" }]
torchvision = [{ index = "pytorch-cpu" }]

[tool.ruff]
target-version = "py38"
Expand Down