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

已于 2022-06-05 09:19:26 修改
阅读量1.1w 收藏 64
点赞数 20
分类专栏： # 深度学习
于 2022-05-17 17:09:30 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/toopoo/article/details/124825357
版权
本文档详细介绍了如何在ARM64架构上安装PyTorch、Torchvision和Torchaudio。首先，建议在虚拟环境中操作，如Anaconda或pyenv。对于PyTorch，可以从PyPI下载匹配Python版本的whl文件并使用pip安装。接着，同样方式安装Torchvision。关于两者对应关系，可查阅官方文档。最后，安装Torchaudio可以直接通过pip完成，它会自动匹配已安装的PyTorch版本。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

文章目录

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

1. 安装pytorch for arm64

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

2. 安装torchvison for arm64

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日的截图，若有更新请移步官网。
在这里插入图片描述
在这里插入图片描述

树莓派安装pytorch arrch64架构
网络架构
04-16 687
如果你的python版本不在支持的列表那么就要自己安装 这里推荐python3.6.13 ,相对稳定些兼容性好。
树莓派4B安装Pytorch, torchvision（附已编译安装包）
Tiiktak‘s
05-12 3588
树莓派4B安装Pytorch, torchvision Install Pytorch Raspberry Pi 4B Linux raspberrypi 4.19.75-v7l+ #1270 SMP Tue Sep 24 18:51:41 BST 2019 armv7l GNU/Linux 2G RAM 16G DISK 增加交换内存 关闭内存交换: sudo dphys-swapfile swapoff 修改配置文件 sudo vim /etc/dphys-swapfile，设置CONF.
在ARM64架构上轻松运行PyTorch：专为Python 3.8打造的深度学习利器
gitblog_09783的博客
10-22 924
在ARM64架构上轻松运行PyTorch：专为Python 3.8打造的深度学习利器 【下载地址】适用于Python3.8版本的PyTorchARM64架构 本仓库提供了专为Python 3.8环境设计的PyTorch库编译版本，特别适配于ARM64（也称为AArch64）架构的系统。这包括但不限于树莓派、Jetson系...
PyTorch中torch、torchtext、torchvision、torchaudio版本对应关系
qq_57438473的博客
11-02 1万+
pytorch相关版本问题
树莓派 4b 最新arm64版 pytorch1.7 whl
11-20
树莓派 4b 最新arm64版 pytorch1.7 whl。请注意这是arm64喂版的，不是32位版的。python版本是3.7
在ARM64架构上轻松运行PyTorch：专为Python 3.8定制
gitblog_09804的博客
10-22 1396
在ARM64架构上轻松运行PyTorch：专为Python 3.8定制 【下载地址】适用于Python3.8版本的PyTorchARM64架构 本仓库提供了专为Python 3.8环境设计的PyTorch库编译版本，特别适配于ARM64（也称为AArch64）架构的系统。这包括但不限于树莓派、Jetson系列开发板或一些...
基于ARM64麒麟V10系统安装pytorch
YYC1503的博客
03-17 4092
基于ARM64麒麟V10系统安装pytorch 本篇基于ARM64架构编译安装pytorch深度学习框架，不支持GPU 安装环境 名称 版本 Linux kylinV10 python 3.7.4 pytorch 1.7.0 离线包准备 需要提前配置好Yum源和准备好相关离线安转包 直接安装 #!/bin/bash yum install python3-pillow -y pip3 install numpy-1.20.1-cp37-cp37m-linux_aarch64.w
torch对应版本torchvision和torchAudio
热门推荐
04-18 2万+
返回False，那么是无法使用gpu的，所以需要找到pytorch和cuda的对应关系，不能安装cpu版本，不然无法使用gpu的。
jetson agx orin安装pytorch失败（未正确安装gpu版本）
12-05
安装CUDA和cuDNN后，再使用pip安装对应版本的PyTorch、torchaudio和torchvision wheel文件。例如，可以使用如下命令： ``` pip3 install torch-2.3.0-cp310-cp310-linux_aarch64.whl pip3 install torchaudio-2.3.0...
pytorch-aarch64:用于 aarch64 ARMv8 ARM64 的 PyTorch 轮 (whl) 和 docker，具有视觉、音频、文本和 csprng
07-24
pytorch-aarch64 用于aarch64 /ARMv8/ARM64设备的 、、、和轮 (whl) 和aarch64图像 ( ) | | 安装 pip install torch -f https://torch.kmtea.eu/whl/stable.html 备份链接： pip install torch -f https://cf.torch.kmtea.eu/whl/stable.html （国内用户请使用 ） 如果需要，添加torchvision 、 torchaudio 、 torchtext 、 torchcsprng和其他包。 注意：此命令安装最新版本。 要选择特定版本，请查看部分。 要手动选择whl文件，请检查。 码头工人 docker run -it kumatea/pytorch 要拉取镜像，请运行docker pull kumatea/p
Jetson AGX ORIN 配置 FGVC-PIM 神经网络（包含 arm64 下面 torch 和 torchvision 配置内容）
weixin_52303102的博客
03-22 798
然后再运行训练数据集的命令，发现可以训练了（这个不打算训练，所以。设置好代码的虚拟环境之后，根据缺少的库的内容，使用。需要的库及其命令如下，不需要找了，可以直接下载。的虚拟环境，之后需要向环境中添加需要的安装包。强行中断了），配置结束！
PyCharm安装pytorch（亲自尝试版！！！！，纯小白都能学会）
GodlenThunder的博客
07-23 6436
小白如何正确安装pytorch，从此踏上炼丹的道路。本文讲解了Pycharm中pytorch库的安装，并对一些可能遇到的问题做出了解答。
python arm64_PyTorch-aarch64
weixin_35657612的博客
03-01 2310
PyTorch源码编译步骤：1、源码编译环境：操作系统：debian 9.12交换空间：1GPython版本：3.5硬件：CPU：RK3399(aarch64)内存：4G2、下载依赖包：下载pytorch及其依赖包时，默认从github上下载，如果网络不好、容易断开时，可在gitee上找到对应包克隆链接，然后修改对应配置文件，进行下载。需要细致耐心。3、编译主要参数：设置最大作业数：export ...
pytorch与torchvision版本对应
nnzzll的博客
05-26 5192
Pytorch与torchvision版本对应
ubuntu16.04-arm版 下载pytorch
weixin_46969696的博客
06-24 2179
要在华为的atlas 200dk上配置一下深度学习的推理环境。 网上编译好的包不能直接用，要重新开始编译。 下载pytorch 对系统进行更新 1.sudo apt update && sudo apt upgrade 创建python的虚拟环境，因为这个架构还不支持conda，所以要用venv创建虚拟环境。 2.python3 -m venv pytorch 如果是指定版本的，将python3改为python3.6,python3.7即可。但是要先执行如下代码。 3.sudo apt-ge
arm-gpu环境中安装pytorch经验总结
w1028900249的博客
04-08 2281
以我安装经验来看，虽然docker里面安装了显卡驱动，但是和实体机器要保持大版本相同，才不会应该cuda版本和docker中pytorch版本不匹配而出错。docker 里面的cuda版本是11.8是可以的，但是docker里面cuda版本12.0会出问题。网站拉取docker，这个页面的docker里面自身不带cuda，如果对实体机器cuda版本有要求。没有提供arm的gpu版本的pytorch安装包，所以安装方式通常有基于pytorch源码编译和基于做好的docker实现。、找到容器对应的配置文件。
install torch for arm
xiaohaier8593的博客
09-10 1238
参考文章 pytorch-aarch64/README.md at main · KumaTea/pytorch-aarch64 · GitHub
aarch64安装pytorch gpu
最新发布
01-14
### 如何在aarch64架构上安装支持GPU的PyTorch #### 安装前准备 为了成功部署适用于aarch64架构并带有GPU加速功能的PyTorch环境，需先确认目标机器已正确设置好CUDA和cuDNN库。对于基于ARM架构的设备而言，这一步骤尤为关键，因为并非所有的预构建二进制文件都兼容此类硬件配置[^3]。 #### CUDA验证 确保本地环境中已经安装了适当版本的CUDA工具链。可以通过命令`nvcc -V`来查询当前系统的CUDA版本号。此操作有助于后续选择相匹配的PyTorch发行版[^5]。 #### Anaconda环境搭建 建议采用Anaconda作为Python包管理器，因为它能简化依赖关系管理和隔离工作区。具体做法如下： 1. 下载适合aarch64平台的Anaconda或Miniconda安装脚本； 2. 创建一个新的Conda虚拟环境，并指定所需的Python版本（例如3.7）； 3. 激活新建的虚拟环境以便继续下一步的操作。 ```bash # 假设选择了Miniconda wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh bash Miniconda3-latest-Linux-aarch64.sh source ~/.bashrc conda create --name pytorch_env python=3.7 conda activate pytorch_env ``` #### PyTorch GPU版本获取途径 鉴于官方渠道可能缺乏直接针对aarch64架构的支持，推荐利用第三方提供的轮子仓库进行安装。这里给出了一种方法是从特定链接下载经过适配后的PyTorch及其配套组件。 ```bash pip install torch==1.8.0+cu102 torchvision==0.9.0+cu102 torchaudio===0.8.0 -f https://download.pytorch.org/whl/torch_stable.html ``` 需要注意的是，在执行上述指令之前，请务必核实所选PyTorch版本与现有CUDA版本之间的兼容性。此外，如果遇到网络连接受限的情况，则可以考虑预先下载对应的.whl文件并通过离线方式完成安装过程。
评论 3
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值