Skip to content

Modify getting started page to add aarch64 builds - #7652

Merged
atalman merged 5 commits into
pytorch:mainfrom
atalman:fix_generate_binary_build_matrix
Jan 15, 2026
Merged

Modify getting started page to add aarch64 builds#7652
atalman merged 5 commits into
pytorch:mainfrom
atalman:fix_generate_binary_build_matrix

Conversation

@atalman

@atalman atalman commented Jan 15, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor
  1. Fixes: On aarch64, pip install torch resulted in the CPU version? pytorch#168065
    Implements this approach: On aarch64, pip install torch resulted in the CPU version? pytorch#168065 (comment)

When stable CUDA is chosen, we display:

For Linux x86 use:
pip3 install torch torchvision

For Linux Aarch64:
pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu128
  1. Remove cuda 12.9 from matrix, since its somehwat limited release.

@vercel

vercel Bot commented Jan 15, 2026

Copy link
Copy Markdown

@atalman is attempting to deploy a commit to the Meta Open Source Team on Vercel.

A member of the Team first needs to authorize it.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jan 15, 2026
@atalman atalman changed the title Modify getting started page Modify getting started page to add aarch64 builds Jan 15, 2026
huydhn
huydhn reviewed Jan 15, 2026
View reviewed changes
Comment thread tools/scripts/generate_binary_build_matrix.py
and gpu_arch_version == STABLE_CUDA_VERSIONS[channel]
and os == LINUX
):
return f"""For Linux x86 use:

@huydhn huydhn Jan 15, 2026

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Should we get a review from @malfet or @albanD here for this section?

huydhn
huydhn approved these changes Jan 15, 2026
View reviewed changes

@huydhn huydhn left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

malfet
malfet approved these changes Jan 15, 2026
View reviewed changes

@malfet malfet left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, though I think generate_binay_build_matrix.py script are somewhat complicated to read atm

@atalman
atalman merged commit 8a59b59 into pytorch:main Jan 15, 2026
91 of 94 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@huydhn huydhn huydhn approved these changes
@malfet malfet malfet approved these changes

Assignees

No one assigned

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

On aarch64, pip install torch resulted in the CPU version?

3 participants

@atalman @huydhn @malfet