Update npu docs (#2745)

PaddlePaddle · Dec 31, 2024 · d7fc2e5 · d7fc2e5
1 parent 0a83956
commit d7fc2e5
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 4 deletions.
diff --git a/docs/other_devices_support/paddlepaddle_install_NPU.en.md b/docs/other_devices_support/paddlepaddle_install_NPU.en.md
@@ -33,6 +33,12 @@ Currently, Python 3.9 wheel installation packages are provided. If you have a ne
 python3.9 -m pip install paddlepaddle==3.0.0.dev20240520 -i https://www.paddlepaddle.org.cn/packages/nightly/cpu/
 python3.9 -m pip install paddle_custom_npu==3.0.0.dev20240719 -i https://www.paddlepaddle.org.cn/packages/nightly/npu/
 ```
+* Set environment variables on the arm machine (not required for x86 environment)
+```bash
+# Solve the error reported by libgomp on the arm machine
+# "libgomp cannot allocate memory in static TLS block"
+export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libgomp.so.1:$LD_PRELOAD
+```
 * After verifying that the installation package is installed, run the following command
 ```bash
 python -c "import paddle; paddle.utils.run_check()"

diff --git a/docs/other_devices_support/paddlepaddle_install_NPU.md b/docs/other_devices_support/paddlepaddle_install_NPU.md
@@ -8,14 +8,14 @@ comments: true
 
 ## 1、docker环境准备
 * 拉取镜像，此镜像仅为开发环境，镜像中不包含预编译的飞桨安装包，镜像中已经默认安装了昇腾算子库 CANN-8.0.T13。
-```
+```bash
 # 适用于 X86 架构
 docker pull registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-x86_64-gcc84-py39
 # 适用于 Aarch64 架构
 docker pull registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-aarch64-gcc84-py39
 ```
 * 参考如下命令启动容器，ASCEND_RT_VISIBLE_DEVICES 指定可见的 NPU 卡号
-```
+```bash
 docker run -it --name paddle-npu-dev -v $(pwd):/work \
     --privileged --network=host --shm-size=128G -w=/work \
     -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
@@ -28,13 +28,19 @@ docker run -it --name paddle-npu-dev -v $(pwd):/work \
 当前提供 Python3.9 的 wheel 安装包。如有其他 Python 版本需求，可以参考[飞桨官方文档](https://www.paddlepaddle.org.cn/install/quick)自行编译安装。
 
 * 下载安装 Python3.9 的 wheel 安装包
-```
+```bash
 # 注意需要先安装飞桨 cpu 版本
 python3.9 -m pip install paddlepaddle==3.0.0.dev20240520 -i https://www.paddlepaddle.org.cn/packages/nightly/cpu/
 python3.9 -m pip install paddle_custom_npu==3.0.0.dev20240719 -i https://www.paddlepaddle.org.cn/packages/nightly/npu/
 ```
-* 验证安装包安装完成之后，运行如下命令
+* arm机器上需要设置环境变量（x86环境无需设置）
+```bash
+# 解决libgomp在arm机器上报错
+# "libgomp cannot allocate memory in static TLS block"
+export LD_PRELOAD=/usr/lib/aarch64-linux-gnu/libgomp.so.1:$LD_PRELOAD
 ```
+* 验证安装包安装完成之后，运行如下命令
+```bash
 python -c "import paddle; paddle.utils.run_check()"
 ```
 预期得到如下输出结果