Merge pull request #51 from daavoo/master

Fix missing "/include/" in cudnn.h search
FloopCZ · Feb 5, 2018 · ba7f78a · ba7f78a
2 parents fbbd76c + 58a0f84
commit ba7f78a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensorflow_cc/cmake/build_tensorflow.sh b/tensorflow_cc/cmake/build_tensorflow.sh
@@ -35,7 +35,7 @@ fi
 if [ -e /opt/cuda/include/cudnn.h ]; then
     echo "Using CUDNN from /opt/cuda"
     export CUDNN_INSTALL_PATH=/opt/cuda
-elif [ -e /usr/local/cuda/cudnn.h ]; then
+elif [ -e /usr/local/cuda/include/cudnn.h ]; then
     echo "Using CUDNN from /usr/local/cuda"
     export CUDNN_INSTALL_PATH=/usr/local/cuda
 elif [ -e /usr/include/cudnn.h ]; then