Skip to content

Commit 08cfc7c

Browse files
authored
Modify installation.md for adding pip extra index of torch-npu (#1272)
### What this PR does / why we need it? Modify installation.md for adding pip extra index of torch-npu ### How was this patch tested? No need --------- Signed-off-by: Icey <1790571317@qq.com>
1 parent e112317 commit 08cfc7c

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

docs/source/installation.md

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -123,10 +123,15 @@ apt update -y
123123
apt install -y gcc g++ cmake libnuma-dev wget git
124124
```
125125

126-
**[Optional]** Config the extra-index of `pip` if you are working on a **x86** machine, so that the torch with cpu could be found:
126+
**[Optional]** Then config the extra-index of `pip` if you are working on a x86 machine or using torch-npu dev version:
127127

128128
```bash
129+
# For x86 machine
129130
pip config set global.extra-index-url https://download.pytorch.org/whl/cpu/
131+
# For torch-npu dev version
132+
pip config set global.extra-index-url https://mirrors.huaweicloud.com/ascend/repos/pypi
133+
# For x86 torch-npu dev version
134+
pip config set global.extra-index-url "https://download.pytorch.org/whl/cpu/ https://mirrors.huaweicloud.com/ascend/repos/pypi"
130135
```
131136

132137
Then you can install `vllm` and `vllm-ascend` from **pre-built wheel**:
@@ -156,7 +161,6 @@ cd ..
156161
# Install vLLM Ascend
157162
git clone --depth 1 --branch |vllm_ascend_version| https://github.com/vllm-project/vllm-ascend.git
158163
cd vllm-ascend
159-
export PIP_EXTRA_INDEX_URL=https://mirrors.huaweicloud.com/ascend/repos/pypi
160164
pip install -v -e .
161165
cd ..
162166
```

0 commit comments

Comments
 (0)