Tensorflow(一)

第一个Graph图

  • Tensorflow程序可以通过tf.device函数来指定运行每一个操作的设备
  • 这个设备可以是本地的CPU或GPU,也可以是某一远程服务器
import tensorflow as tf

'''
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参数来打印运行每一个计算的设备
with tf.device('/cpu:0')  安装了gpu版本的tensorflow的话with块内的在CPU执行,with块外的在GPU执行
GPU版本的tensorflow在PC上安装的话,需要NVIDIA的显卡,并且显卡支持cuda
除了可以看到最后的计算结果之外,还可以看到类似"add: /job:localhost/replica:0/task:0/cpu:0/"这样的输出
这些输出显示了执行的每一个运算的设备。比如加法操作add是通过CPU来运行的,因为它的设备名称中包含了/cpu:0
在配置好GPU环境的Tensorflow中,如果操作没有明确的指定运行设备,那么Tensorflow会优先选择GPU来执行代码
'''

with tf.device('/
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值