问题现象描述

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

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐