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>