Skip to content

Commit b5202d2

Browse files
authored
Revert "Fix gather_runners_info script on ROCm (#6799)" (#6805)
This reverts commit 007e89b. This is the first time I see a memory error with pip https://github.com/pytorch/pytorch/actions/runs/15767425131/job/44447322381
1 parent b18a93a commit b5202d2

File tree

1 file changed

+1
-10
lines changed
  • .github/actions/upload-benchmark-results

1 file changed

+1
-10
lines changed

.github/actions/upload-benchmark-results/action.yml

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -18,16 +18,7 @@ runs:
1818
shell: bash
1919
run: |
2020
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
21+
python3 -mpip install boto3==1.35.33 psutil==7.0.0 pynvml==12.0.0 torch
3122
3223
- name: Check that GITHUB_TOKEN is defined
3324
env:

0 commit comments

Comments
 (0)