From 04671c18b0be375a0c3f394b1ada22409e33f845 Mon Sep 17 00:00:00 2001
From: ronnywang <524019753@qq.com>
Date: Tue, 28 Dec 2021 13:19:08 +0800
Subject: [PATCH] add paddle dcu doc (#5441)
---
tutorials/mobilenetv3_prod/Step6/README.md | 4 ++++
.../linux_dcu_train_infer_python/README.md | 22 +++++++++++++++----
2 files changed, 22 insertions(+), 4 deletions(-)
diff --git a/tutorials/mobilenetv3_prod/Step6/README.md b/tutorials/mobilenetv3_prod/Step6/README.md
index fc91a559..b6a560a9 100644
--- a/tutorials/mobilenetv3_prod/Step6/README.md
+++ b/tutorials/mobilenetv3_prod/Step6/README.md
@@ -157,6 +157,8 @@ python3 train.py --data-path=./ILSVRC2012 --lr=0.1 --batch-size=256
[Epoch 1, iter: 4790] top1: 0.08750, top5: 0.24531, lr: 0.01000, loss: 5.28853, avg_reader_cost: 0.05164 sec, avg_batch_cost: 0.06852 sec, avg_samples: 64.0, avg_ips: 934.08427 images/sec.
```
+DCU运行需要设置环境变量 `export HIP_VISIBLE_DEVICES=0`,启动命令与Linux GPU完全相同。
+
* 单机多卡训练
```bash
@@ -166,6 +168,8 @@ python3 -m paddle.distributed.launch --gpus="0,1,2,3" train.py --data-path="./IL
更多配置参数可以参考[train.py](./train.py)的`get_args_parser`函数。
+DCU运行需要设置环境变量 `export HIP_VISIBLE_DEVICES=0,1,2,3`,启动命令与Linux GPU完全相同。
+
### 4.2 模型评估
diff --git a/tutorials/tipc/linux_dcu_train_infer_python/README.md b/tutorials/tipc/linux_dcu_train_infer_python/README.md
index 9e024cc7..66f5d2dd 100644
--- a/tutorials/tipc/linux_dcu_train_infer_python/README.md
+++ b/tutorials/tipc/linux_dcu_train_infer_python/README.md
@@ -4,8 +4,22 @@
- [1. 简介](#1)
- [2. Linux DCU 基础训练推理功能开发与规范](#2)
- - [2.1 开发流程](#2.1)
- - [2.2 核验点](#2.2)
- [3. Linux DCU 基础训练推理测试开发与规范](#3)
- - [3.1 开发流程](#3.1)
- - [3.2 核验点](#3.2)
+
+
+
+## 1. 简介
+
+Linux DCU 基础训练推理开发过程主要步骤与[《Linux GPU/CPU 基础训练推理开发》](../train_infer_python/README.md)一致。启动命令与Linux GPU相同,只需修改环境变量 `CUDA_VISIBLE_DEVICES` 为 `HIP_VISIBLE_DEVICES`。
+
+
+
+## 2. Linux DCU 基础训练推理功能开发与规范
+
+参考[《Linux GPU/CPU 基础训练推理开发》](../train_infer_python/README.md)。
+
+
+
+## 3. Linux DCU 基础训练推理测试开发与规范
+
+参考[《Linux GPU/CPU 基础训练推理测试开发规范》](../train_infer_python/test_train_inference_python.md)。
--
GitLab