Ver Fonte

update npu docker and modify infer problem (#3418)

a31413510 há 9 meses atrás
pai
commit
948a0a2dd8

+ 2 - 2
docs/other_devices_support/paddlepaddle_install_DCU.en.md

@@ -10,7 +10,7 @@ Currently, PaddleX supports Haiguang Z100 series chips. Considering environmenta
 Pull the image. Note that this image is only for development environments and does not include pre-compiled PaddlePaddle installation packages.
 
 ```bash
-docker pull registry.baidubce.com/device/paddle-dcu:dtk23.10.1-kylinv10-gcc73-py310
+docker pull ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle-dcu:dtk24.04.1-kylinv10-gcc82
 ```
 
 Start the container with the following command as a reference:
@@ -19,7 +19,7 @@ Start the container with the following command as a reference:
 docker run -it --name paddle-dcu-dev -v `pwd`:/work \
   -w=/work --shm-size=128G --network=host --privileged  \
   --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
-  registry.baidubce.com/device/paddle-dcu:dtk23.10.1-kylinv10-gcc73-py310 /bin/bash
+  ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle-dcu:dtk24.04.1-kylinv10-gcc82 /bin/bash
 ```
 
 ## 2. Install PaddlePaddle Package

+ 2 - 2
docs/other_devices_support/paddlepaddle_install_DCU.md

@@ -10,7 +10,7 @@ comments: true
 拉取镜像,此镜像仅为开发环境,镜像中不包含预编译的飞桨安装包
 
 ```
-docker pull registry.baidubce.com/device/paddle-dcu:dtk23.10.1-kylinv10-gcc73-py310
+docker pull ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle-dcu:dtk24.04.1-kylinv10-gcc82
 ```
 参考如下命令启动容器
 
@@ -18,7 +18,7 @@ docker pull registry.baidubce.com/device/paddle-dcu:dtk23.10.1-kylinv10-gcc73-py
 docker run -it --name paddle-dcu-dev -v `pwd`:/work \
   -w=/work --shm-size=128G --network=host --privileged  \
   --cap-add=SYS_PTRACE --security-opt seccomp=unconfined \
-  registry.baidubce.com/device/paddle-dcu:dtk23.10.1-kylinv10-gcc73-py310 /bin/bash
+  ccr-2vdh3abv-pub.cnc.bj.baidubce.com/paddlepaddle/paddle-dcu:dtk24.04.1-kylinv10-gcc82 /bin/bash
 ```
 
 ## 2、安装paddle包

+ 10 - 7
docs/other_devices_support/paddlepaddle_install_NPU.en.md

@@ -7,12 +7,12 @@ comments: true
 Currently, PaddleX supports the Ascend 910B chip (more models are under support. If you have a related need for other models, please submit an issue to inform us). The Ascend driver version is 23.0.3. Considering the differences in environments, we recommend using the <b>Ascend development image provided by PaddlePaddle</b> to complete the environment preparation.
 
 ## 1. Docker Environment Preparation
-* Pull the image. This image is only for the development environment and does not contain a pre-compiled PaddlePaddle installation package. The image has CANN-8.0.T13, the Ascend operator library, installed by default.
+* Pull the image. This image is only for the development environment and does not contain a pre-compiled PaddlePaddle installation package. The image has CANN-8.0.RC2, the Ascend operator library, installed by default.
 ```bash
 # For X86 architecture
-docker pull registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-x86_64-gcc84-py39
+docker pull ccr-2vdh3abv-pub.cnc.bj.baidubce.com/device/paddle-npu:cann80RC2-ubuntu20-npu-base-x86_64-gcc84
 # For Aarch64 architecture
-docker pull registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-aarch64-gcc84-py39
+docker pull ccr-2vdh3abv-pub.cnc.bj.baidubce.com/device/paddle-npu:cann80RC2-ubuntu20-npu-base-aarch64-gcc84
 ```
 * Start the container with the following command. ASCEND_RT_VISIBLE_DEVICES specifies the visible NPU card numbers.
 ```bash
@@ -22,17 +22,20 @@ 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-$(uname -m)-gcc84-py39 /bin/bash
+    ccr-2vdh3abv-pub.cnc.bj.baidubce.com/device/paddle-npu:cann80RC2-ubuntu20-$(uname -m)-gcc84 /bin/bash
 ```
 ## 2. Install Paddle Package
-Currently, Python 3.9 wheel installation packages are provided. If you have a need for other Python versions, you can refer to the [PaddlePaddle official documentation](https://www.paddlepaddle.org.cn/en/install/quick) to compile and install them yourself.
-
-* Download and install the Python 3.9 wheel installation package
+* Download and install the Python wheel installation package
 ```bash
 # Note: You need to install the CPU version of PaddlePaddle first
 python -m pip install paddlepaddle -i https://www.paddlepaddle.org.cn/packages/nightly/cpu
 python -m pip install paddle-custom-npu -i https://www.paddlepaddle.org.cn/packages/nightly/npu
 ```
+* CANN-8.0.RC2 does not support some versions of numpy and opencv, it is recommended to install the specified versions.
+```bash
+python -m pip install numpy==1.26.4
+python -m pip install opencv-python==3.4.18.65
+```
 * Set environment variables on the arm machine (not required for x86 environment)
 ```bash
 # Solve the error reported by libgomp on the arm machine

+ 10 - 7
docs/other_devices_support/paddlepaddle_install_NPU.md

@@ -7,12 +7,12 @@ comments: true
 当前 PaddleX 支持昇腾 910B 芯片(更多型号还在支持中,如果您有其他型号的相关需求,请提交issue告知我们),昇腾驱动版本为 23.0.3。考虑到环境差异性,我们推荐使用<b>飞桨官方提供的昇腾开发镜像</b>完成环境准备。
 
 ## 1、docker环境准备
-* 拉取镜像,此镜像仅为开发环境,镜像中不包含预编译的飞桨安装包,镜像中已经默认安装了昇腾算子库 CANN-8.0.T13
+* 拉取镜像,此镜像仅为开发环境,镜像中不包含预编译的飞桨安装包,镜像中已经默认安装了昇腾算子库 CANN-8.0.RC2
 ```bash
 # 适用于 X86 架构
-docker pull registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-x86_64-gcc84-py39
+docker pull ccr-2vdh3abv-pub.cnc.bj.baidubce.com/device/paddle-npu:cann80RC2-ubuntu20-npu-base-x86_64-gcc84
 # 适用于 Aarch64 架构
-docker pull registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-aarch64-gcc84-py39
+docker pull ccr-2vdh3abv-pub.cnc.bj.baidubce.com/device/paddle-npu:cann80RC2-ubuntu20-npu-base-aarch64-gcc84
 ```
 * 参考如下命令启动容器,ASCEND_RT_VISIBLE_DEVICES 指定可见的 NPU 卡号
 ```bash
@@ -22,17 +22,20 @@ 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-$(uname -m)-gcc84-py39 /bin/bash
+    ccr-2vdh3abv-pub.cnc.bj.baidubce.com/device/paddle-npu:cann80RC2-ubuntu20-$(uname -m)-gcc84 /bin/bash
 ```
 ## 2、安装paddle包
-当前提供 Python3.9 的 wheel 安装包。如有其他 Python 版本需求,可以参考[飞桨官方文档](https://www.paddlepaddle.org.cn/install/quick)自行编译安装。
-
-* 下载安装 Python3.9 的 wheel 安装包
+* 下载安装 wheel 安装包
 ```bash
 # 注意需要先安装飞桨 cpu 版本
 python -m pip install paddlepaddle -i https://www.paddlepaddle.org.cn/packages/nightly/cpu
 python -m pip install paddle-custom-npu -i https://www.paddlepaddle.org.cn/packages/nightly/npu
 ```
+* CANN-8.0.RC2 对 numpy 和 opencv 部分版本不支持,建议安装指定版本
+```bash
+python -m pip install numpy==1.26.4
+python -m pip install opencv-python==3.4.18.65
+```
 * arm机器上需要设置环境变量(x86环境无需设置)
 ```bash
 # 解决libgomp在arm机器上报错