Skip to content

Commit 3d38da7

Browse files
committed
update
1 parent 07d465f commit 3d38da7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.github/workflows/_ascend_npu_benchmark.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -138,6 +138,10 @@ jobs:
138138
torch==${{ steps.list-torch-version.outputs.torch-version }} \
139139
numpy==1.*
140140
141+
- name: Install nightly torchvision and torchaudio
142+
run: |
143+
pip install --pre torchvision torchaudio --no-deps --index-url https://download.pytorch.org/whl/nightly/cpu
144+
141145
- name: Install dependencies for all the models
142146
run: |
143147
python benchmark/install.py --userbenchmark test_bench --continue_on_fail

0 commit comments

Comments
 (0)