Hi,
You can install PyTorch with the instruction shared in this topic.
For example,
PyTorch v1.7:
$ wget https://nvidia.box.com/shared/static/cs3xn3td6sfgtene6jdvsxlr366m2dhq.whl -O torch-1.7.0-cp36-cp36m-linux_aarch64.whl
$ sudo apt-get install python3-pip libopenblas-base libopenmpi-dev
$ pip3 install Cython
$ pip3 install numpy torch-1.7.0-cp36-cp36m-linux_aarch64.whl
Torchaudio v0.7.0:
$ git clone --branch release/0.7 https://github.com/pytorch/audio torchaudio
$ cd torchaudio
$ export BUILD_VERSION=0.7.0
$ sudo python3 setup.py install
Thanks.