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 b18a93a commit b5202d2Copy full SHA for b5202d2
.github/actions/upload-benchmark-results/action.yml
@@ -18,16 +18,7 @@ runs:
18
shell: bash
19
run: |
20
set -eux
21
-
22
- python3 -mpip install boto3==1.35.33 psutil==7.0.0 pynvml==12.0.0
23
- # NB: I'm using PyTorch here to get the device name, however, it needs to
24
- # install the correct version of PyTorch manually for now. Any PyTorch
25
- # version is fine, I just use 2.7.1 to satify PYPIDEP linter
26
- if command -v nvidia-smi; then
27
- python3 -mpip install torch==2.7.1
28
- elif command -v amd-smi; then
29
- python3 -mpip install torch==2.7.1 --index-url https://download.pytorch.org/whl/rocm6.3
30
- fi
+ python3 -mpip install boto3==1.35.33 psutil==7.0.0 pynvml==12.0.0 torch
31
32
- name: Check that GITHUB_TOKEN is defined
33
env:
0 commit comments