Better way of detecting CUDA support in pip-found TF.

tradr-project · Apr 20, 2020 · 8527529 · 8527529
1 parent 2be03b7
commit 8527529
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cmake/detect_tf_pip.cmake b/cmake/detect_tf_pip.cmake
@@ -154,10 +154,10 @@ if(NOT EXISTS ${TENSORFLOW_LIBRARY2})
 endif()
 
 execute_process(
-    COMMAND ldd ${TENSORFLOW_LIBRARY}
-    OUTPUT_VARIABLE LDD_OUTPUT
+    COMMAND nm -CD ${TENSORFLOW_ORIG_LIBRARY}
+    OUTPUT_VARIABLE NM_OUTPUT
 )
-if(${LDD_OUTPUT} MATCHES "libcudart.so")
+if(${NM_OUTPUT} MATCHES "cudaError")
   set(HAS_TENSORFLOW_GPU 1)
   message("-- -- The Tensorflow library is compiled with CUDA support.")
 else()