Skip to content

Commit 1c62487

Browse files
ZhangYangclaude
andcommitted
ci: introduce uv to accelerate pip install
Replace pip3 install with uv pip install in _build-and-test.yml. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
1 parent 836ecfb commit 1c62487

1 file changed

Lines changed: 9 additions & 4 deletions

File tree

.github/workflows/_build-and-test.yml

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,10 @@ jobs:
2626
image: ${{ inputs.image }}
2727
env:
2828
HOME: /root
29+
UV_INDEX_URL: "https://mirrors.huaweicloud.com/repository/pypi/simple"
30+
UV_INDEX_STRATEGY: "unsafe-best-match"
31+
UV_NO_CACHE: 1
32+
UV_SYSTEM_PYTHON: 1
2933
volumes:
3034
- /usr/local/dcmi:/usr/local/dcmi
3135
- /usr/local/bin/npu-smi:/usr/local/bin/npu-smi
@@ -43,19 +47,20 @@ jobs:
4347
- name: Compile torch
4448
working-directory: /root/build/pytorch/pytorch
4549
run: |
46-
pip3 install -r requirements.txt -i https://mirrors.huaweicloud.com/repository/pypi/simple
50+
pip3 install uv
51+
uv pip install -r requirements.txt
4752
export _GLIBCXX_USE_CXX11_ABI=1
4853
export USE_CUDA=0
4954
export USE_XNNPACK=0
5055
python setup.py develop
5156
- name: Compile and install torch_npu
5257
working-directory: /root/build/npu/pytorch
5358
run: |
54-
pip3 install -r requirements.txt -i https://mirrors.huaweicloud.com/repository/pypi/simple
59+
uv pip install -r requirements.txt
5560
bash ci/build.sh --python=3.8
56-
pip3 install dist/torch_npu*.whl
61+
uv pip install dist/torch_npu*.whl
5762
- name: Do the test
5863
working-directory: /root/build
5964
run: |
60-
pip3 install -r npu/pytorch/test/requirements.txt -i https://mirrors.huaweicloud.com/repository/pypi/simple --no-deps
65+
uv pip install -r npu/pytorch/test/requirements.txt --no-deps
6166
python npu/pytorch/ci/access_control_test.py

0 commit comments

Comments
 (0)