install nvidia-driver in build_wheels_linux

Author: NicolasHug

.github/workflows/build_wheels_linux.yml

@@ -184,6 +184,8 @@ jobs:
           # When building RC, set the version to be the current candidate version,
           # otherwise, leave it alone so nightly will pick up the latest
           echo "PYTORCH_VERSION=${{ matrix.stable_version }}" >> "${GITHUB_ENV}"
+      - name: Install nvidia driver
+        uses: pytorch/test-infra/.github/actions/setup-nvidia@main
       - uses: ./test-infra/.github/actions/setup-binary-builds
         env:
           PLATFORM: ${{ inputs.architecture == 'aarch64'  && 'linux-aarch64' || ''}}