test libcudnn - conv2d in tensorflow/keras uses cudnn

cudnn_test.py

import tensorflow as tf
config=tf.ConfigProto(
    log_device_placement=True,
)
sess = tf.Session(config=config)
assert tf.test.is_gpu_available()
assert tf.test.is_built_with_cuda()
op = tf.shape(
    tf.nn.conv2d(tf.random_normal([1,10,10,10]),
                 tf.random_normal([2,10,10,10]),
                 strides=[1, 2, 2, 1],
                 padding='SAME',
                 use_cudnn_on_gpu=True
                )
)
with tf.Session() as sess:
    result = sess.run(op)
    print(result)