Tensorflow学习笔记十五——加速计算

1.Tensorflow单机实现

import tensorflow as tf

a=tf.Variable(tf.constant([1.0,shape=[1]),name="a")
b=tf.Variable(tf.constant([3.0,shape=[1]),name="b")
result=a+b
init_op=tf.global_variables_initializer()

with tf.Session(config=tf.COnfigProto(log_device_placement=True)) as sess:
    sess.run(init_op)
    print(result)

strategy = tf.distribute.MirroredStragegy()
print('Number of devices: { }'.format(strategy.num_replocas_in_sync))

import tensorflow.compat.v1 as tf

with tf.device("/cpu:0"):
    a=tf.Variable(tf.constant([1.0,shape=[1]),name="a")
    b=tf.Variable(tf.constant([3.0,shape=[1]),name="b")
with tf.device("gpu"0):
    result=a+b
init_op=tf.global_variables_initializer()

with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
    sess.run(init_op)
    print(sess.run(result))