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

文章目录

  • 1. 安装pytorch for arm64
  • 2. 安装torchvison for arm64
  • 3. pytorch和torchvison的对应关系(即时更新)
  • 4. 安装torchaudio for arm64
  • 5. pytorch以及torchaudio的版本对应关系

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

1. 安装pytorch for arm64

  • 下载
    点击 torch · PyPI,下载对应的版本,例如
    torch-1.11.0-cp38-cp38-manylinux2014_aarch64.whl
    其中cp38代表python 3.8
    aarch64arm64
  • 安装
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的对应关系(即时更新)

如果寻找pytorchtorchvison的对应关系呢?
官网链接:
pytorch/vision: Datasets, Transforms and Models specific to Computer Vision
以下是2022年5月17日的截图,若有更新请移步官网。
arm64下安装pytorch,torchvision,torchaudio以及它们之间的版本对应关系_第1张图片

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日的截图,若有更新请移步官网。
arm64下安装pytorch,torchvision,torchaudio以及它们之间的版本对应关系_第2张图片
arm64下安装pytorch,torchvision,torchaudio以及它们之间的版本对应关系_第3张图片

你可能感兴趣的:(#,深度学习)