Преглед на файлове

add npu arm docker image (#1907)

a31413510 преди 1 година
родител
ревизия
a64847100d
променени са 2 файла, в които са добавени 10 реда и са изтрити 5 реда
  1. 6 4
      docs/tutorials/INSTALL_OTHER_DEVICES.md
  2. 4 1
      docs/tutorials/base/devices_use_guidance.md

+ 6 - 4
docs/tutorials/INSTALL_OTHER_DEVICES.md

@@ -6,8 +6,10 @@
 - 1. 拉取镜像,此镜像仅为开发环境,镜像中不包含预编译的飞桨安装包,镜像中已经默认安装了昇腾算子库 CANN-8.0.T13。
 
 ```
-# 适用于 X86 架构,暂时不提供 Arch64 架构镜像
+# 适用于 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
 ```
 
 - 2. 参考如下命令启动容器,ASCEND_RT_VISIBLE_DEVICES 指定可见的 NPU 卡号
@@ -18,7 +20,7 @@ docker run -it --name paddle-npu-dev -v $(pwd):/work \
     -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
     -v /usr/local/dcmi:/usr/local/dcmi \
     -e ASCEND_RT_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" \
-    registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-x86_64-gcc84-py39 /bin/bash
+    registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-$(uname -m)-gcc84-py39 /bin/bash
 ```
 ### 1.2 安装 paddle 包
 当前提供 Python3.9 的 wheel 安装包。如有其他 Python 版本需求,可以参考[飞桨官方文档](https://www.paddlepaddle.org.cn/install/quick)自行编译安装。
@@ -27,8 +29,8 @@ docker run -it --name paddle-npu-dev -v $(pwd):/work \
 
 ```
 # 注意需要先安装飞桨 cpu 版本
-pip install https://paddle-whl.bj.bcebos.com/paddlex/npu/2024_07_23/paddlepaddle-0.0.0-cp39-cp39-linux_x86_64.whl
-pip install https://paddle-whl.bj.bcebos.com/paddlex/npu/2024_07_23/paddle_custom_npu-0.0.0-cp39-cp39-linux_x86_64.whl
+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/
 ```
 - 2. 验证安装包
 安装完成之后,运行如下命令。

+ 4 - 1
docs/tutorials/base/devices_use_guidance.md

@@ -35,7 +35,10 @@ git clone https://gitee.com/paddlepaddle/PaddleX.git
 ```bash
 cd PaddleX
 
-# 注意,如果是 arm64 架构的 xpu 机器,需要先手动安装 tool_helpers,x86 架构机器不需要
+# 注意,如果是 arm64 架构的机器,需要先手动安装 tool_helpers,x86 架构机器不需要
+# python3.9 版本,适用 npu arm64 架构机器
+pip install https://paddle-model-ecology.bj.bcebos.com/paddlex/whl/paddlenlp-device/tool_helpers-0.1.1-cp39-cp39-linux_aarch64.whl
+# python3.10 版本,适用 xpu arm64 架构机器
 pip install https://paddle-model-ecology.bj.bcebos.com/paddlex/whl/paddlenlp-device/tool_helpers-0.1.1-cp310-cp310-linux_aarch64.whl
 
 # 安装 PaddleX whl