[docker] update npu docker (#10479)

This commit is contained in:
xvxuopop
2026-05-13 20:56:43 +08:00
committed by GitHub
parent ca50f22c38
commit 6b9df75ab9
5 changed files with 9 additions and 6 deletions

View File

@@ -38,7 +38,7 @@ jobs:
cancel-in-progress: ${{ github.ref != 'refs/heads/main' }}
container:
image: ascendai/cann:8.3.rc2-910b-ubuntu22.04-py3.11
image: ascendai/cann:9.0.0-910b-ubuntu22.04-py3.11
env:
HF_ENDPOINT: https://hf-mirror.com
HF_TOKEN: ${{ secrets.HF_TOKEN }}
@@ -65,8 +65,8 @@ jobs:
- name: Install dependencies
run: |
uv venv
uv pip install -r requirements/npu.txt
uv pip install -e .
uv pip install -r requirements/npu.txt
uv pip install -r requirements/dev.txt
- name: Install node
@@ -89,5 +89,7 @@ jobs:
make build
- name: Test with pytest
shell: bash
run: |
source /usr/local/Ascend/ascend-toolkit/set_env.sh
make test