Link to tf.test.is_built_with_cuda

tensorflow · Apr 6, 2019 · 0ec1786 · 0ec1786
1 parent ca2ddb5
commit 0ec1786
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow/python/framework/test_util.py b/tensorflow/python/framework/test_util.py
@@ -1332,7 +1332,8 @@ def is_gpu_available(cuda_only=False, min_cuda_compute_capability=None):
   """Returns whether TensorFlow can access a GPU.
 
   Warning: if not GPU version of the package is installed, the function would
-  also, return False.
+  also, return False. Use `tf.test.is_built_with_cuda` to validate if TensorFlow
+  was build with CUDA support.
 
   Args:
     cuda_only: limit the search to CUDA GPUs.