|
|
@@ -3,11 +3,11 @@
|
|
|
## 1. 昇腾 NPU 飞桨安装
|
|
|
### 1.1 环境准备
|
|
|
当前 PaddleX 支持昇腾 910B 芯片(更多型号还在支持中,如果您有其他型号的相关需求,请提交issue告知我们),昇腾驱动版本为 23.0.3。考虑到环境差异性,我们推荐使用飞桨官方提供的标准镜像完成环境准备。
|
|
|
-- 1. 拉取镜像,此镜像仅为开发环境,镜像中不包含预编译的飞桨安装包,镜像中已经默认安装了昇腾算子库 CANN-8.0.RC1。
|
|
|
+- 1. 拉取镜像,此镜像仅为开发环境,镜像中不包含预编译的飞桨安装包,镜像中已经默认安装了昇腾算子库 CANN-8.0.T13。
|
|
|
|
|
|
```
|
|
|
# 适用于 X86 架构,暂时不提供 Arch64 架构镜像
|
|
|
-docker pull registry.baidubce.com/device/paddle-npu:cann80RC1-ubuntu20-x86_64-gcc84-py39
|
|
|
+docker pull registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-x86_64-gcc84-py39
|
|
|
```
|
|
|
|
|
|
- 2. 参考如下命令启动容器,ASCEND_RT_VISIBLE_DEVICES 指定可见的 NPU 卡号
|
|
|
@@ -18,7 +18,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:cann80RC1-ubuntu20-x86_64-gcc84-py39 /bin/bash
|
|
|
+ registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-x86_64-gcc84-py39 /bin/bash
|
|
|
```
|
|
|
### 1.2 安装 paddle 包
|
|
|
当前提供 Python3.9 的 wheel 安装包。如有其他 Python 版本需求,可以参考[飞桨官方文档](https://www.paddlepaddle.org.cn/install/quick)自行编译安装。
|
|
|
@@ -27,8 +27,8 @@ docker run -it --name paddle-npu-dev -v $(pwd):/work \
|
|
|
|
|
|
```
|
|
|
# 注意需要先安装飞桨 cpu 版本
|
|
|
-pip install https://paddle-whl.bj.bcebos.com/paddlex/npu/paddlepaddle-0.0.0-cp39-cp39-linux_x86_64.whl
|
|
|
-pip install https://paddle-whl.bj.bcebos.com/paddlex/npu/paddle_custom_npu-0.0.0-cp39-cp39-linux_x86_64.whl
|
|
|
+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
|
|
|
```
|
|
|
- 2. 验证安装包
|
|
|
安装完成之后,运行如下命令。
|