文章目录
首先强烈建议在虚拟环境下安装!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日的截图,若有更新请移步官网。