Fix dependencies bugs

Change: 116925769

1 files changed, 7 insertions, 1 deletions

diff --git a/configure b/configure
index 3f0a067eae..2d7ec77aec 100755
--- a/configure
+++ b/configure
@@ -99,12 +99,18 @@ while true; do
   else
     TF_CUDNN_EXT=".$TF_CUDNN_VERSION"
   fi
-  if [ -e "$CUDNN_INSTALL_PATH/libcudnn.so${CUDNNEXT}" -o -e "$CUDNN_INSTALL_PATH/lib64/libcudnn.so${TF_CUDNN_EXT}" ]; then
+  if [ -e "$CUDNN_INSTALL_PATH/libcudnn.so${TF_CUDNN_EXT}" -o -e "$CUDNN_INSTALL_PATH/lib64/libcudnn.so${TF_CUDNN_EXT}" ]; then
+    break
+  fi
+  CUDNN_PATH_FROM_LDCONFIG="$(ldconfig -p | sed -n 's/.*libcudnn.so .* => \(.*\)/\1/p')"
+  if [ -e "${CUDNN_PATH_FROM_LDCONFIG}${TF_CUDNN_EXT}" ]; then
+    CUDNN_INSTALL_PATH="$(dirname ${CUDNN_PATH_FROM_LDCONFIG})"
     break
   fi
   echo "Invalid path to cuDNN ${TF_CUDNN_VERSION} toolkit. Neither of the following two files can be found:"
   echo "$CUDNN_INSTALL_PATH/lib64/libcudnn.so${TF_CUDNN_EXT}"
   echo "$CUDNN_INSTALL_PATH/libcudnn.so${TF_CUDNN_EXT}"
+  echo "${CUDNN_PATH_FROM_LDCONFIG}${TF_CUDNN_EXT}"
   if [ -z "$fromuser" ]; then
     exit 1
   fi