没有合适的资源?快使用搜索试试~ 我知道了~
jupyter notebook tensorflow打印device信息实例
0 下载量 134 浏览量
2020-09-17
13:00:23
上传
评论
收藏 34KB PDF 举报
温馨提示
试读
1页
主要介绍了jupyter notebook tensorflow打印device信息实例,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
资源推荐
资源详情
资源评论
jupyter notebook tensorflow打印打印device信息实例信息实例
主要介绍了jupyter notebook tensorflow打印device信息实例,具有很好的参考价值,希望对大家有所帮助。一
起跟随小编过来看看吧
juypter notebook中直接使用log_device_placement=True打印不出来device信息
# Creates a graph.
with tf.device('/device:CPU:0'):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True,allow_soft_placement=True))
# Runs the op.
print(sess.run(c))
需要使用output_partition_graphs来输出device信息
# Creates a graph.
with tf.device('/device:GPU:0'):
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
# Creates a session with log_device_placement set to True.
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True,allow_soft_placement=True))
# Runs the op.
options = tf.RunOptions(output_partition_graphs=True)
metadata = tf.RunMetadata()
c_val = sess.run(c, options=options, run_metadata=metadata)
print metadata.partition_graphs
补充知识:补充知识:Jupyter无法在控制台打印无法在控制台打印
因为数据有中文,所以我特意在jupter前面设置了
reload(sys)
sys.setdefaultencoding("utf-8")
结果使用print语句的时候无法输入内容。究其原因,是因为reload的时候把sdout变为ipython的对象了,所以要临时储存一下
stdout的对象。
不妨试一试以下代码
import sys
stdo = sys.stdout
reload(sys)
sys.setdefaultencoding('utf-8')
sys.stdout= stdo
以上这篇jupyter notebook tensorflow打印device信息实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希
望大家多多支持我们。
资源评论
weixin_38687199
- 粉丝: 4
- 资源: 924
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功