Ignore Nvidia drivers that fail to install in all_drivers_test.sh.

Some drivers fail to build on our Ubuntu VMs due to
NVIDIA/open-gpu-kernel-modules#594.

PiperOrigin-RevId: 634085739

tools/gpu/all_drivers_test.sh

@@ -24,7 +24,20 @@ trap "rm -f ${tmp_file}" EXIT
 make sudo TARGETS=tools/gpu:main ARGS="list --outfile=${tmp_file}"
 read -r -a versions <<< "$(cat "${tmp_file}")"
 
+num_successful=0
 for driver in "${versions[@]}"; do
+  set +e
   make sudo TARGETS=tools/gpu:main ARGS="install --version ${driver}"
+  install_exit_code=$?
+  set -e
+  if [[ $install_exit_code -ne 0 ]]; then
+    echo "Installing driver ${driver} failed. Not testing this version." >&2
+    continue
+  fi
   make gpu-smoke-tests RUNTIME_ARGS="--debug"
+  num_successful="$(( $num_successful + 1 ))"
 done
+if [[ "$num_successful" == 0 ]]; then
+  echo 'No version was successfully tested.' >&2
+  exit 1
+fi