Skip to content

Commit c034ac2

Browse files
authored
Use setup-nvidia-windows from test-infra (#1426)
1 parent b0f6ee5 commit c034ac2

1 file changed

Lines changed: 1 addition & 6 deletions

File tree

.github/workflows/windows_cuda_wheel.yaml

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -91,12 +91,7 @@ jobs:
9191
uses: actions/checkout@v6
9292

9393
- name: Update NVIDIA driver
94-
shell: cmd
95-
run: |
96-
curl --retry 3 -kL https://ossci-windows.s3.amazonaws.com/580.88-data-center-tesla-desktop-win10-win11-64bit-dch-international.exe --output driver_installer.exe
97-
start /wait driver_installer.exe -s -noreboot
98-
del driver_installer.exe
99-
nvidia-smi
94+
uses: pytorch/test-infra/.github/actions/setup-nvidia-windows@main
10095

10196
- name: Remove src/ folder
10297
run: bash packaging/remove_src.sh

0 commit comments

Comments
 (0)