博客

测试TensorFlow GPU是否安装成功

  • Post author:
  • Post category:其他


import tensorflow as tf 
sess = tf.Session() 
a = tf.constant(1) 
b = tf.constant(2) 
print(sess.run(a+b))

结果: 3