| commit | 2806fa4470c17f8ce8159b7516eccc4a91496290 | [log] [tgz] |
|---|---|---|
| author | Huy Do <huydhn@gmail.com> | Wed Mar 29 21:14:27 2023 +0000 |
| committer | PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> | Wed Mar 29 21:14:27 2023 +0000 |
| tree | 8846b4c5daf6703763de97a9281693dd9d8e6418 | |
| parent | b93e1f377ecd775750fca43ecebe6eb54abb6efa [diff] |
Use the latest NVIDIA driver from setup-nvidia (#97840) This goes with https://github.com/pytorch/test-infra/pull/3949 Pull Request resolved: https://github.com/pytorch/pytorch/pull/97840 Approved by: https://github.com/ZainRizvi
diff --git a/.github/workflows/_linux-test.yml b/.github/workflows/_linux-test.yml index 44dd294..53fb23f 100644 --- a/.github/workflows/_linux-test.yml +++ b/.github/workflows/_linux-test.yml
@@ -92,8 +92,6 @@ - name: Install nvidia driver, nvidia-docker runtime, set GPU_FLAG uses: pytorch/test-infra/.github/actions/setup-nvidia@main if: contains(inputs.build-environment, 'cuda') && !contains(matrix.config, 'nogpu') - with: - driver-version: "525.85.05" - name: Lock NVIDIA A100 40GB Frequency run: |