Skip to content

Commit 51f64a1

Browse files
committed
Update docs
1 parent 0f29499 commit 51f64a1

File tree

1 file changed

+24
-1
lines changed

1 file changed

+24
-1
lines changed

README_EN.md

Lines changed: 24 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ For detailed instructions on installing PaddlePaddle, please refer to [Installat
7373

7474
### 4. Install dependencies
7575

76-
#### Method 1: One-click installation (recommended)
76+
#### Method 1: One-click installation (recommended on GPU / CPU)
7777

7878
Run the following command to automatically install all necessary dependencies:
7979
```
@@ -88,6 +88,29 @@ Detailed [installation]((https://www.paddlepaddle.org.cn/install/quick?docurl=/d
8888
8989
> If you wish to train and infer using **bf16**, please use a GPU that supports **bf16**, such as the A100.
9090
91+
#### Method 3: For Ascend 910B
92+
93+
PaddleMIX currently supports the Ascend 910B chip (more chip types are still being supported. If you have related needs for other chip types, please submit a question and let us know). Our recommended driver version is 23.0.3. Considering the differences in environments, we recommend using docker image provided by PaddlePaddle to complete the environment preparation.
94+
95+
* Refer to the following command to start the container, ASCEND_RT_VISIBLE_DEVICES specifies the visible NPU card number
96+
97+
```shell
98+
docker run -it --name paddle-npu-dev -v $(pwd):/work \
99+
--privileged --network=host --shm-size=128G -w=/work \
100+
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
101+
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
102+
-v /usr/local/dcmi:/usr/local/dcmi \
103+
-e ASCEND_RT_VISIBLE_DEVICES="0,1,2,3,4,5,6,7" \
104+
registry.baidubce.com/device/paddle-npu:cann80T13-ubuntu20-$(uname -m)-gcc84-py39 /bin/bash
105+
```
106+
107+
* Install PaddlePaddle in the container
108+
109+
```shell
110+
python3.9 -m pip install paddlepaddle==3.0.0.dev20240520 -i https://www.paddlepaddle.org.cn/packages/nightly/cpu/
111+
python3.9 -m pip install paddle_custom_npu==3.0.0.dev20240719 -i https://www.paddlepaddle.org.cn/packages/nightly/npu/
112+
```
113+
91114

92115
## Tutorial
93116

0 commit comments

Comments
 (0)