美文网首页
first_graph

first_graph

作者: Lonelyroots | 来源:发表于2023-01-30 21:21 被阅读0次

    first_graph

    import tensorflow as tf
    
    # tf.Variable生成的变量,每次迭代都会变化,
    # 这个变量也就是我们要去计算的结果,所以说要计算什么,就把什么定义为Variable
    '''
    TensorFlow程序可以通过tf.device函数来指定运行每一个操作的设备。 
     
    这个设备可以是本地的CPU或者GPU,也可以是某一台远程的服务器。 
    TensorFlow会给每一个可用的设备一个名称,tf.device函数可以通过设备的名称,来指定执行运算的设备。比如CPU在TensorFlow中的名称为/cpu:0。 
     
    在默认情况下,即使机器有多个CPU,TensorFlow也不会区分它们,所有的CPU都使用/cpu:0作为名称。 
     
    –而一台机器上不同GPU的名称是不同的,第n个GPU在TensorFlow中的名称为/gpu:n。 
    –比如第一个GPU的名称为/gpu:0,第二个GPU名称为/gpu:1,以此类推。 
    –TensorFlow提供了一个快捷的方式,来查看运行每一个运算的设备。 
    –在生成会话时,可以通过设置log_device_placement参数来打印运行每一个运算的设备。 
    
    –除了可以看到最后的计算结果之外,还可以看到类似“add: /job:localhost/replica:0/task:0/cpu:0”这样的输出 
    –这些输出显示了执行每一个运算的设备。比如加法操作add是通过CPU来运行的,因为它的设备名称中包含了/cpu:0。 
    –在配置好GPU环境的TensorFlow中,如果操作没有明确地指定运行设备,那么TensorFlow会优先选择GPU
    '''
    x = tf.Variable(3,name='x')
    print(x)
    y = tf.Variable(4,name='y')
    print(y)
    f = x*x*y + y + 2 
    print(f)
    
    # 创建一个计算图的一个上下文环境
    # 配置里面是把具体运行过程在哪里执行给打印出来
    sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
    # 碰到session.run()就会立刻去调用计算
    sess.run(x.initializer)     # 变量需要初始化
    sess.run(y.initializer)
    result = sess.run(f)
    print(result)
    sess.close()
    

    文章到这里就结束了!希望大家能多多支持Python(系列)!六个月带大家学会Python,私聊我,可以问关于本文章的问题!以后每天都会发布新的文章,喜欢的点点关注!一个陪伴你学习Python的新青年!不管多忙都会更新下去,一起加油!

    Editor:Lonelyroots

    相关文章

      网友评论

          本文标题:first_graph

          本文链接:https://www.haomeiwen.com/subject/hfwphdtx.html