pytorch arm平台, aarch64 pip 快速安装

本文介绍在ARM设备上安装PyTorch的困难与解决方案,推荐使用KumaTea编译的whl包,步骤包括选择对应版本和CPU型号,以及如何从kmtea.eu下载并安装。

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

pytorch 官方并未提供 arm平台的安装包.
于是按照官方页面提供的命令是无论如何也无法安装成功的.
只要你的CPU用的ARM芯片. 那么无论你是什么操作系统, Liunx, windows,maxos 都不行
解决方法有2个,
第一个是下载源代码然后进行手动编译.(此方法太过复杂和耗时, 没点耐心搞不好.)
第二个是用社区其它人编译好的版本. 这个方法就非常简单了.
请参考: https://github.com/KumaTea/pytorch-aarch64
得感谢 KumaTea

使用方法也很简单, 到pytorch官方界面生成命令. 然后把 -f 后面的网址改成

-f https://torch.kmtea.eu/whl/stable.html

意思是, 安装包到 https://torch.kmtea.eu/whl/stable.html 去找.
例如我的安装命令是

pip3 install torch==1.6.0 torchvision==0.7.0 torchaudio==0.9.0 -f https://torch.kmtea.eu/whl/stable-cn.html

1.注意版本,

2.还有+cpu 要去掉,

真不知道该怎么选命令, 可以自己用浏览器打开

https://torch.kmtea.eu/whl/stable-cn.html

然后选择自己对应的whl文件, 用迅雷下载好之后,用pip命令安装, 其它的大包也可以用迅雷下载, 速度贼快.

pip install ***你的包文件名.whl  进行安装, 速度快又稳定. 
评论 14
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值