From d272cac299e9d9876534c4d5dbe9dece697f0b30 Mon Sep 17 00:00:00 2001 From: pengzirong Date: Fri, 1 Jul 2022 09:30:57 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E4=B8=8B=E8=BD=BD=E5=AF=B9?= =?UTF-8?q?=E5=BA=94=E5=88=86=E6=94=AF=E5=91=BD=E4=BB=A4;=E5=88=A0?= =?UTF-8?q?=E9=99=A4=E9=83=A8=E5=88=86=E7=8E=AF=E5=A2=83=E5=8F=98=E9=87=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.zh.md | 36 ++++++------------- ...11\350\243\205\346\214\207\345\215\227.md" | 2 +- 2 files changed, 11 insertions(+), 27 deletions(-) diff --git a/README.zh.md b/README.zh.md index 59ea542893a..5513d145dbe 100644 --- a/README.zh.md +++ b/README.zh.md @@ -40,16 +40,20 @@ pip3 install wheel ## 编译安装PyTorch和昇腾插件 -首先安装官方torch包，然后编译安装插件。（ARM架构CPU安装，请参见第一条FAQ） +首先安装官方torch包，然后编译安装插件。 ```sh -pip3 install torch==1.8.1+cpu +#x86_64 +pip3 install torch==1.8.1+cpu #安装cpu版本PyTorch 下载地址：（https://download.pytorch.org/whl/torch） + +#aarch64 +#社区未提供arm架构cpu安装包，请参见FAQ第一条，使用源码编译安装pytorch ``` 编译生成pytorch插件的二进制安装包。 ``` -git clone -b master https://gitee.com/ascend/pytorch.git & cd pytorch # 下载master分支代码，进入插件根目录 +git clone -b v1.8.1-3.0.rc2 https://gitee.com/ascend/pytorch.git & cd pytorch # 下载v1.8.1-3.0.rc2分支代码，进入插件根目录 # 指定python版本编包方式： bash ci/build.sh --python=3.7 # 或 @@ -78,23 +82,16 @@ source env.sh ## 自定义环境变量 -以下环境变量为NPU场景下使用的功能类或可以提升性能的环境变量： - -``` -export TASK_QUEUE_ENABLE=1 # 使用异步任务下发，异步调用acl接口，建议默认开启，开启设置为1 -``` - 可选的环境变量可能会对运行的模型产生影响: ``` -export DYNAMIC_COMPILE_ENABLE=1 # 动态shape特性功能，针对shape变化场景，可选，开启设置为1 export COMBINED_ENABLE=1 # 非连续两个算子组合类场景优化，可选，开启设置为1 export ACL_DUMP_DATA=1 # 算子数据dump功能，调试时使用，可选，开启设置为1 -export DYNAMIC_OP="ADD#MUL" # 算子实现，ADD和MUL算子在不同场景下有不同的性能表现。可选 ``` **表 1** 环境变量说明 + - - - - - - - - -

配置项

说明

@@ -132,11 +129,6 @@ export DYNAMIC_OP="ADD#MUL" # 算子实现，ADD和MUL算子在不同场景下

aicpu算子包路径。

TASK_QUEUE_ENABLE

-

使用异步任务下发，异步调用acl接口。建议开启，开启设置为1。

-

HCCL_WHITELIST_DISABLE

配置在使用HCCL时是否开启通信白名单。

@@ -169,11 +161,6 @@ export DYNAMIC_OP="ADD#MUL" # 算子实现，ADD和MUL算子在不同场景下
  • 0：不开启Event日志。
  • 1：开启Event日志。
  • 其他值为非法值。

DYNAMIC_COMPILE_ENABLE

-

（可选）动态shape特性功能，针对shape变化场景，开启设置为1（PyTorch1.8.1不支持该环境变量）。

-

COMBINED_ENABLE

（可选）非连续两个算子组合类场景优化，开启设置为1。

@@ -184,11 +171,6 @@ export DYNAMIC_OP="ADD#MUL" # 算子实现，ADD和MUL算子在不同场景下

（可选）算子数据dump功能，调试时使用，开启设置为1。

DYNAMIC_OP

-

（可选）算子实现，ADD和MUL算子在不同场景下有不同的性能表现。默认不设置。

-

unset GOMP_CPU_AFFINITY

（可选）当系统为openEuler及其继承操作系统时，如UOS，需设置此命令，取消CPU绑核。

@@ -197,6 +179,8 @@ export DYNAMIC_OP="ADD#MUL" # 算子实现，ADD和MUL算子在不同场景下
+ + ## 执行单元测试脚本 验证运行, 输出结果OK diff --git "a/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" "b/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" index 7f76e3ee285..f08664232bc 100644 --- "a/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" +++ "b/docs/zh/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227/PyTorch\345\256\211\350\243\205\346\214\207\345\215\227.md" @@ -68,7 +68,7 @@ pip3 install torch==1.8.1+cpu #安装cpu版本PyTorch 下载地址：（https:// 编译生成pytorch插件的二进制安装包。 ``` -git clone -b master https://gitee.com/ascend/pytorch.git & cd pytorch # 下载master分支代码，进入插件根目录 +git clone -b v1.8.1-3.0.rc2 https://gitee.com/ascend/pytorch.git & cd pytorch # 下载master分支代码，进入插件根目录 # 指定python版本编包方式： bash ci/build.sh --python=3.7 # 或 -- Gitee