OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
master
pytorch/.ci/aarch64_linux
History
Huy Do 4400c5d31e Continue to build nightly CUDA 12.9 for internal (#163029) 
Revert part of https://github.com/pytorch/pytorch/pull/161916 to continue building CUDA 12.9 nightly

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163029
Approved by: https://github.com/malfet
2025-10-11 08:26:47 +00:00
..
aarch64_ci_build.sh Continue to build nightly CUDA 12.9 for internal (#163029) 2025-10-11 08:26:47 +00:00
aarch64_ci_setup.sh Revert "[build] pin setuptools>=77 to enable PEP 639 (#158104)" 2025-07-21 22:46:53 +00:00
aarch64_wheel_ci_build.py Build and Install Arm Compute Library in manylinux docker image (#159737) 2025-10-01 11:33:51 +00:00
build_aarch64_wheel.py Build and Install Arm Compute Library in manylinux docker image (#159737) 2025-10-01 11:33:51 +00:00
embed_library.py [CD] Move linux-aarch64 build scripts (#139815) 2024-11-06 04:16:48 +00:00
README.md [CD] Move linux-aarch64 build scripts (#139815) 2024-11-06 04:16:48 +00:00

README.md

Aarch64 (ARM/Graviton) Support Scripts

Scripts for building aarch64 PyTorch PIP Wheels. These scripts build the following wheels:

  • torch
  • torchvision
  • torchaudio
  • torchtext
  • torchdata

Aarch64_ci_build.sh

This script is design to support CD operations within PyPi manylinux aarch64 container, and be executed in the container. It prepares the container and then executes aarch64_wheel_ci_build.py to build the wheels. The script "assumes" the PyTorch repo is located at: /pytorch and will put the wheels into /artifacts.

Usage

DESIRED_PYTHON=<PythonVersion> aarch64_ci_build.sh

NOTE: CI build is currently EXPERMINTAL

Build_aarch64_wheel.py

This app allows a person to build using AWS EC3 resources and requires AWS-CLI and Boto3 with AWS credentials to support building EC2 instances for the wheel builds. Can be used in a codebuild CD or from a local system.

Usage

build_aarch64_wheel.py --key-name <YourPemKey> --use-docker --python 3.8 --branch <RCtag>