Ascend Extension for PyTorch Installation
导言
参与到 Ascend/pytorch项目里,首先就是如何安装。
由于黄区严格的网络限制,docker难以使用,还是只能conda下重新安装
肯定要是昇腾的设备
前提条件¶
安装配套版本的CANN软件,并使用其脚本
前辈装好了,直接用这部分
安装PyTorch框架¶
已有conda环境t00906153_bindCore
中包含torch 2.1.0
通过源码安装torch_npu插件¶
参考文档
# CentOS
yum install -y patch libjpeg-turbo-devel dos2unix openblas git
yum install -y gcc==7.3.0 cmake==3.12.0
git clone -b v2.1.0-6.0.rc1 https://gitee.com/ascend/pytorch.git
cd pytorch
# build 前要source pytorch环境
source . {xxx}/ascend-toolkit/set_env.sh
bash ci/build.sh --python=3.8
pip3 install --upgrade dist/torch_npu-2.1.0.post3-cp38-cp38-linux_aarch64.whl
反思:为什么第一次编译失败了
- 惯性思维:不严谨,想当然了,clone了master来运行
- 文档阅读不细致,就急于尝试:公司里有人写wiki是很难得的,如果有wiki可以参考,一定先看完,可以避免很多坑。
- 编译时缺头文件,如果不在项目代码里,就很有可能是相应的系统包没有安装。