Skip to content

Modify getting started page to add aarch64 builds #7652

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

Jump to bottom
Merged
atalman merged 5 commits into from
Jan 15, 2026
Merged

Modify getting started page to add aarch64 builds #7652

atalman merged 5 commits into from
Jan 15, 2026

Conversation

@atalman
Copy link
Contributor

@atalman atalman commented Jan 15, 2026 •
edited
Loading

  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
Copy link

vercel bot commented Jan 15, 2026

@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
tools/scripts/generate_binary_build_matrix.py
and gpu_arch_version == STABLE_CUDA_VERSIONS[channel]
and os == LINUX
):
return f"""For Linux x86 use:
Copy link
Contributor

@huydhn huydhn Jan 15, 2026

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
Copy link
Contributor

@huydhn huydhn left a comment

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
Copy link
Contributor

@malfet malfet left a comment

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