跳转至

Ascend Extension for PyTorch Installation

导言

参与到 Ascend/pytorch项目里,首先就是如何安装。

由于黄区严格的网络限制,docker难以使用,还是只能conda下重新安装

肯定要是昇腾的设备

前提条件

安装配套版本的CANN软件,并使用其脚本

source . {xxx}/ascend-toolkit/set_env.sh

前辈装好了,直接用这部分

安装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

反思:为什么第一次编译失败了

  1. 惯性思维:不严谨,想当然了,clone了master来运行
  2. 文档阅读不细致,就急于尝试:公司里有人写wiki是很难得的,如果有wiki可以参考,一定先看完,可以避免很多坑。
  3. 编译时缺头文件,如果不在项目代码里,就很有可能是相应的系统包没有安装。

参考文献

评论