[ci] More in NVIDIA driver info dump (#4495)

diff --git a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh
index d8e008f..8baebde 100755
--- a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh
+++ b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build.sh
@@ -31,8 +31,9 @@
 python3 -c 'import tensorflow as tf; print(tf.__version__)'
 
 # Print NVIDIA GPU information inside the docker
-nvidia-smi
 dmesg | grep NVRM
+dpkg -l | grep nvidia
+nvidia-smi || true
 
 ./build_tools/kokoro/gcp_ubuntu/check_vulkan.sh
 
diff --git a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh
index 359857e..e5a1589 100755
--- a/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh
+++ b/build_tools/kokoro/gcp_ubuntu/cmake-bazel/linux/x86-turing/build_kokoro.sh
@@ -29,8 +29,9 @@
 source "${KOKORO_ARTIFACTS_DIR?}/github/iree/build_tools/kokoro/gcp_ubuntu/docker_common.sh"
 
 # Print NVIDIA GPU information inside the VM
-nvidia-smi
 dmesg | grep NVRM
+dpkg -l | grep nvidia
+nvidia-smi || true
 
 # Sets DOCKER_RUN_ARGS
 docker_setup
diff --git a/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-turing/build.sh b/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-turing/build.sh
index 29f70c5..75d27e1 100755
--- a/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-turing/build.sh
+++ b/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-turing/build.sh
@@ -30,8 +30,9 @@
 python3 --version
 
 # Print NVIDIA GPU information inside the docker
-nvidia-smi
 dmesg | grep NVRM
+dpkg -l | grep nvidia
+nvidia-smi || true
 
 ./build_tools/kokoro/gcp_ubuntu/check_vulkan.sh
 
diff --git a/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-turing/build_kokoro.sh b/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-turing/build_kokoro.sh
index b71e84a..542283d 100755
--- a/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-turing/build_kokoro.sh
+++ b/build_tools/kokoro/gcp_ubuntu/cmake/linux/x86-turing/build_kokoro.sh
@@ -29,8 +29,9 @@
 source "${KOKORO_ARTIFACTS_DIR?}/github/iree/build_tools/kokoro/gcp_ubuntu/docker_common.sh"
 
 # Print NVIDIA GPU information inside the VM
-nvidia-smi
 dmesg | grep NVRM
+dpkg -l | grep nvidia
+nvidia-smi || true
 
 # Sets DOCKER_RUN_ARGS
 docker_setup