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
</

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

被折叠的 条评论
为什么被折叠?



