TensorFlow: change cuda-diagnostics to search for so.1

Vijay Vasudevan · Vijay Vasudevan · commit f600f4d82de7 · 2016-02-19T09:08:24.000-08:00
Change: 115010103
diff --git a/tensorflow/stream_executor/cuda/cuda_diagnostics.cc b/tensorflow/stream_executor/cuda/cuda_diagnostics.cc
@@ -165,7 +165,7 @@ port::StatusOr<DriverVersion> Diagnostician::FindDsoVersion() {
   // DSO and yields its version number into the callback data, when found.
   auto iterate_phdr =
       [](struct dl_phdr_info *info, size_t size, void *data) -> int {
-    if (strstr(info->dlpi_name, "libcuda.so")) {
+    if (strstr(info->dlpi_name, "libcuda.so.1")) {
       VLOG(1) << "found DLL info with name: " << info->dlpi_name;
       char resolved_path[PATH_MAX] = {0};
       if (realpath(info->dlpi_name, resolved_path) == nullptr) {
