5a5ac98918
seeing ` File "/usr/local/lib/python3.12/site-packages/torch/__init__.py", line 411, in <module>
from torch._C import * # noqa: F403
^^^^^^^^^^^^^^^^^^^^^^
ImportError: libcufile.so.0: cannot open shared object file: No such file or directory` with arm cu128 nightly.
related to https://github.com/pytorch/pytorch/pull/148137
need to copy the dependency for arm build as well
Pull Request resolved: https://github.com/pytorch/pytorch/pull/148465
Approved by: https://github.com/atalman, https://github.com/abhilash1910
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>