test.py

#Test file for tensorflow operations

import tensorflow as tf
'''
#sets up the graph
node1 = tf.constant(3.0, dtype=tf.float32)
node2 = tf.constant(4.0) # also tf.float32 implicitly
print(node1, node2)

#make a tensor flow session to run the graph

sess = tf.Session()

print (sess.run([node1, node2]))
'''

with tf.device('/gpu:0'):
    a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
    b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
    c = tf.matmul(a, b)

    with tf.Session() as sess:
        print (sess.run(c))