tensorflow 最简单gpu程序

import tensorflow as tf
import numpy as np

with tf.compat.v1.Session() as sess:
    with tf.device('/gpu:0'):
        a = tf.compat.v1.placeholder(tf.int32)
        b = tf.compat.v1.placeholder(tf.int32)
        add = tf.add(a, b)
        sum = sess.run(add, feed_dict={a: 3, b: 4})
        print(sum)