import tensorflow as tf with tf.device('/cpu:0'): a=tf.constant([1.0,2.0,3.0],shape=[3],name='a') b=tf.constant([1.0,2.0,3.0],shape=[3],name='b') with tf.device('/gpu:0'): c=a+b sess=tf.Session(config=tf.ConfigProto(log_device_placement=True)) print(sess.run(c))
查看GPU
最新推荐文章于 2023-12-10 18:20:44 发布