arm64下安装pytorch,torchvision,torchaudio以及它们之间的版本对应关系

首先强烈建议在虚拟环境下安装！anaconda或者pyenv皆可。

1. 安装pytorch for arm64

• 下载
  点击 torch · PyPI，下载对应的版本，例如
  torch-1.11.0-cp38-cp38-manylinux2014_aarch64.whl
  其中cp38代表python 3.8
  aarch64即arm64
• 安装

pip install torch-1.11.0-cp38-cp38-manylinux2014_aarch64.whl

2. 安装torchvison for arm64

• 下载
  下载页面在这里：torchvision · PyPI
  对应上面torch-1.11.0版本的是torchvision-0.12.0。所以需要下载
  torchvision-0.12.0-cp38-cp38-manylinux2014_aarch64.whl
• 安装

pip install torchvision-0.12.0-cp38-cp38-manylinux2014_aarch64.whl

3. pytorch和torchvison的对应关系(即时更新)

如果寻找pytorch和torchvison的对应关系呢？
官网链接：
pytorch/vision: Datasets, Transforms and Models specific to Computer Vision
以下是2022年5月17日的截图，若有更新请移步官网。

4. 安装torchaudio for arm64

安装好torch/torchvision之后，直接使用pip安装即可。

$ pip install torchaudio
Collecting torchaudio
  Downloading torchaudio-0.11.0-cp38-cp38-manylinux2014_aarch64.whl (3.3 MB)
     |████████████████████████████████| 3.3 MB 1.5 MB/s
Requirement already satisfied: torch in /home/ubuntu/.pyenv/versions/3.8.10/envs/v831/lib/python3.8/site-packages (from torchaudio) (1.11.0)
Requirement already satisfied: typing-extensions in /home/ubuntu/.pyenv/versions/3.8.10/envs/v831/lib/python3.8/site-packages (from torch->torchaudio) (4.2.0)
Installing collected packages: torchaudio
Successfully installed torchaudio-0.11.0

5. pytorch以及torchaudio的版本对应关系

官网链接：
pytorch/audio: Data manipulation and transformation for audio signal processing, powered by PyTorch
以下是2022年6月5日的截图，若有更新请移步官网。