install nvidia-driver in build_wheels_linux

Author: NicolasHug

.github/workflows/build_wheels_linux.yml

@@ -217,6 +217,8 @@ jobs:
           source "${BUILD_ENV_FILE}"
           # shellcheck disable=SC2086
           ${CONDA_RUN} ${PIP_INSTALL_TORCH} ${{ inputs.pip-install-torch-extra-args }}
+      - name: Install nvidia driver
+        uses: pytorch/test-infra/.github/actions/setup-nvidia@main
       - name: Run Pre-Script with Caching
         if: ${{ inputs.pre-script != '' }}
         uses: ./test-infra/.github/actions/run-script-with-cache