Update docs to fix torchvision version specifiers for CPU wheels#17380
Update docs to fix torchvision version specifiers for CPU wheels#17380appleparan wants to merge 1 commit intoastral-sh:mainfrom
Conversation
The CPU index provides wheels with different local version suffixes: `+cpu` for x86_64 (Linux) and AMD64 (Windows), but no suffix for aarch64 (Linux) and arm64 (macOS). Add platform-specific version specifiers to match the available wheels. Add environments constraint to avoid resolution conflicts between dependencies and sources markers.
appleparan
force-pushed
the
appleparan/fix-pytorch-docs
branch
from
7312133 to
f80e84e
Compare
|
I've added Without the |
…nflicts Add explicit environments constraint to prevent uv from having resolution conflicts between dependencies and sources markers. This ensures proper resolution for all supported platform/architecture combinations: - Linux x86_64 - Linux aarch64 - macOS arm64 - Windows AMD64 Also clean up trailing whitespace in dependency-groups. Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
| !!! note | ||
|
|
||
| The CPU index provides wheels with different local version suffixes depending on the platform: | ||
| `+cpu` for x86_64 (Linux) and AMD64 (Windows), but no suffix for aarch64 (Linux) and arm64 (macOS). | ||
| As such, `torchvision` requires platform-specific version specifiers to match the available wheels. | ||
|
|
There was a problem hiding this comment.
CC @atalman just to confirm that this is intentional
There was a problem hiding this comment.
For context, that is the list of wheels for torch 2.9.1 on Python 3.14:
torch-2.9.1+cpu-cp314-cp314-manylinux_2_28_aarch64.whl
torch-2.9.1+cpu-cp314-cp314-manylinux_2_28_x86_64.whl
torch-2.9.1+cpu-cp314-cp314-win_amd64.whl
torch-2.9.1+cpu-cp314-cp314t-manylinux_2_28_aarch64.whl
torch-2.9.1+cpu-cp314-cp314t-manylinux_2_28_x86_64.whl
torch-2.9.1+cpu-cp314-cp314t-win_amd64.whl
torch-2.9.1+cu126-cp314-cp314-manylinux_2_28_aarch64.whl
torch-2.9.1+cu126-cp314-cp314-manylinux_2_28_x86_64.whl
torch-2.9.1+cu126-cp314-cp314-win_amd64.whl
torch-2.9.1+cu126-cp314-cp314t-manylinux_2_28_aarch64.whl
torch-2.9.1+cu126-cp314-cp314t-manylinux_2_28_x86_64.whl
torch-2.9.1+cu126-cp314-cp314t-win_amd64.whl
torch-2.9.1+cu128-cp314-cp314-manylinux_2_28_aarch64.whl
torch-2.9.1+cu128-cp314-cp314-manylinux_2_28_x86_64.whl
torch-2.9.1+cu128-cp314-cp314-win_amd64.whl
torch-2.9.1+cu128-cp314-cp314t-manylinux_2_28_aarch64.whl
torch-2.9.1+cu128-cp314-cp314t-manylinux_2_28_x86_64.whl
torch-2.9.1+cu128-cp314-cp314t-win_amd64.whl
torch-2.9.1+cu129-cp314-cp314-manylinux_2_28_aarch64.whl
torch-2.9.1+cu129-cp314-cp314-manylinux_2_28_x86_64.whl
torch-2.9.1+cu129-cp314-cp314t-manylinux_2_28_aarch64.whl
torch-2.9.1+cu129-cp314-cp314t-manylinux_2_28_x86_64.whl
torch-2.9.1+cu130-cp314-cp314-manylinux_2_28_aarch64.whl
torch-2.9.1+cu130-cp314-cp314-manylinux_2_28_x86_64.whl
torch-2.9.1+cu130-cp314-cp314-win_amd64.whl
torch-2.9.1+cu130-cp314-cp314t-manylinux_2_28_aarch64.whl
torch-2.9.1+cu130-cp314-cp314t-manylinux_2_28_x86_64.whl
torch-2.9.1+cu130-cp314-cp314t-win_amd64.whl
torch-2.9.1+rocm6.3-cp314-cp314-manylinux_2_28_x86_64.whl
torch-2.9.1+rocm6.3-cp314-cp314t-manylinux_2_28_x86_64.whl
torch-2.9.1+rocm6.4-cp314-cp314-manylinux_2_28_x86_64.whl
torch-2.9.1+rocm6.4-cp314-cp314t-manylinux_2_28_x86_64.whl
torch-2.9.1+xpu-cp314-cp314-linux_x86_64.whl
torch-2.9.1+xpu-cp314-cp314-win_amd64.whl
torch-2.9.1+xpu-cp314-cp314t-linux_x86_64.whl
torch-2.9.1+xpu-cp314-cp314t-win_amd64.whl
torch-2.9.1-cp314-cp314-macosx_11_0_arm64.whl
torch-2.9.1-cp314-cp314t-macosx_11_0_arm64.whl
There was a problem hiding this comment.
I thought this was fixed in pytorch/test-infra#7589, but perhaps there hasn't been a release since then?
There was a problem hiding this comment.
Agreed. I'm looking for build log for that.
Note: This should be closed also if this PR is closed
There was a problem hiding this comment.
This is the same issue as: pytorch/vision#9249 and its fix for the release 0.25.0 of torchvision.
It's a temporary issue in 0.24.x, so I should close this PR.
|
I close this PR due to this comment
It turns out this is a 0.24.x-specific issue that is fixed in 0.25.0. I will reopen this PR if we need to document this behavior for 0.24.x. |
|
Thank you @appleparan! |
3 participants
Summary
The CPU index provides wheels with different local version suffixes:
+cpufor x86_64 (Linux) and AMD64 (Windows), but no suffix for aarch64 (Linux) and arm64 (macOS). Add platform-specific version specifiers to match the available wheels.This PR doesn't fix a uv bug, but it addresses and can close the issues raised here. #16386 #17314