Useful testing functions after setting up the environment:
import tensorflow as tf
firsttf.test.gpu_device_name()
should gives the correct GPU card name and driver info.tf.test.is_built_with_cuda()
tells if current TF is compiled supporting CUDA.
Some official Tested build configurations.
It tells the compatibility among tensorflow
, cuDNN
, and CUDA
(the package name cudatoolkit
).