README_Ascend_NPU_Acceleration_zh_CN.md 2.47 KB
Newer Older
1
2
3
4
# Ascend NPU 加速

## 简介

5
本文档介绍如何在 Ascend NPU 上使用 MinerU。本文档内容已在`华为 Atlas 800T A2`服务器上测试通过。
6
7
8
```
CPU:鲲鹏 920 aarch64 2.6GHz
NPU:Ascend 910B 64GB
9
10
11
OS:openEuler 22.03 (LTS-SP3)/ Ubuntu 22.04.5 LTS
CANN:8.0.RC2
驱动版本:24.1.rc2.1
12
13
14
```
由于适配 Ascend NPU 的环境较为复杂,建议使用 Docker 容器运行 MinerU。

15
16
通过docker运行MinerU前需确保物理机已安装支持CANN 8.0.RC2的驱动和固件。

17
18
19
20

## 构建镜像
请保持网络状况良好,并执行以下代码构建镜像。    
```bash
21
wget https://gcore.jsdelivr.net/gh/opendatalab/MinerU@master/docker/ascend_npu/Dockerfile -O Dockerfile
22
23
24
25
26
27
28
29
docker build -t mineru_npu:latest .
```
如果构建过程中未发生报错则说明镜像构建成功。


## 运行容器

```bash
30
docker run -it -u root --name mineru-npu --privileged=true \
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
    --ipc=host \
    --network=host \
    --device=/dev/davinci0 \
    --device=/dev/davinci1 \
    --device=/dev/davinci2 \
    --device=/dev/davinci3 \
    --device=/dev/davinci4 \
    --device=/dev/davinci5 \
    --device=/dev/davinci6 \
    --device=/dev/davinci7 \
    --device=/dev/davinci_manager \
    --device=/dev/devmm_svm \
    --device=/dev/hisi_hdc \
    -v /var/log/npu/:/usr/slog \
    -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
    -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
    mineru_npu:latest \
    /bin/bash -c "echo 'source /opt/mineru_venv/bin/activate' >> ~/.bashrc && exec bash"

magic-pdf --help
```


## 已知问题

56
57
- paddleocr使用内嵌onnx模型,仅在默认语言配置下能以较快速度对中英文进行识别
- 自定义lang参数时,paddleocr速度会存在明显下降情况
58
- layout模型使用layoutlmv3时会发生间歇性崩溃,建议使用默认配置的doclayout_yolo模型
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
- 表格解析仅适配了rapid_table模型,其他模型可能会无法使用


## 高性能模式

- 在特定硬件环境可以通过插件开启高性能模式,整体速度相比默认模式提升300%以上

| 系统要求           | 版本/型号        |
|----------------|--------------|
| 芯片类型           | 昇腾910B       |
| CANN版本         | CANN 8.0.RC2 |
| 驱动版本           | 24.1.rc2.1   |
| magic-pdf 软件版本 | \> = 1.2.0   |

- 高性能插件需满足一定的硬件条件和资质要求,如需申请使用请填写以下表单[MinerU高性能版本合作申请表](https://aicarrier.feishu.cn/share/base/form/shrcnb10VaoNQB8kQPA8DEfZC6d)