CPU架构为ARM架构时,由于社区未提供ARM架构CPU版本的torch包,无法使用PIP3命令安装PyTorch1.8.1,需要使用源码编译安装
2023/06/05
65
问题信息
问题来源 | 产品大类 | 产品子类 | 关键字 |
---|---|---|---|
官方 | 安装部署 | PyTorch | 安装PyTorch |
问题现象描述
CPU架构为ARM架构时,无法使用PIP3命令安装PyTorch1.8.1。
原因分析
社区未提供ARM架构CPU版本的torch包。
解决措施
使用源码编译安装。
- 下载PyTorch v1.8.1源码包。
git clone -b v1.8.1 https://github.com/pytorch/pytorch.git --depth=1 pytorch_v1.8.1
- 进入源码包获取被动依赖代码。
cd pytorch_v1.8.1git submodule syncgit submodule update --init --recursive
- 配置环境变量。
export USE_XNNPACK=0
- 执行编译安装。
python3 setup.py install
本页内容