问题现象描述
CPU架构为ARM架构时,无法使用PIP3命令安装PyTorch1.8.1。
原因分析
社区未提供ARM架构CPU版本的torch包。
解决措施
使用源码编译安装。
1.下载PyTorch v1.8.1源码包。
git clone -b v1.8.1 https://github.com/pytorch/pytorch.git --depth=1 pytorch_v1.8.1
2.进入源码包获取被动依赖代码。
cd pytorch_v1.8.1git submodule syncgit submodule update --init --recursive
3.配置环境变量。
export USE_XNNPACK=0
4.执行编译安装。
python3 setup.py install
当面临ARM架构上缺少PyTorch1.8.1预编译包的问题时,可以采用源码编译的方式进行安装。步骤包括:克隆v1.8.1源码、更新子模块、设置环境变量和执行Python安装脚本。
3669

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



