We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 07d465f commit 3d38da7Copy full SHA for 3d38da7
.github/workflows/_ascend_npu_benchmark.yml
@@ -138,6 +138,10 @@ jobs:
138
torch==${{ steps.list-torch-version.outputs.torch-version }} \
139
numpy==1.*
140
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
+
145
- name: Install dependencies for all the models
146
run: |
147
python benchmark/install.py --userbenchmark test_bench --continue_on_fail
0 commit comments