jupyter notebook tensorflow打印device信息实例


Posted in Python onApril 20, 2020

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信息实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
使用python将mdb数据库文件导入postgresql数据库示例
Feb 17 Python
python使用正则搜索字符串或文件中的浮点数代码实例
Jul 11 Python
Python的Bottle框架中实现最基本的get和post的方法的教程
Apr 30 Python
python实现按长宽比缩放图片
Jun 07 Python
python3使用flask编写注册post接口的方法
Dec 28 Python
Python----数据预处理代码实例
Mar 20 Python
Python使用itchat模块实现简单的微信控制电脑功能示例
Aug 26 Python
Python中 Global和Nonlocal的用法详解
Jan 20 Python
使用Python爬虫库requests发送表单数据和JSON数据
Jan 25 Python
Python使用pdb调试代码的技巧
May 03 Python
如何用python写个模板引擎
Jan 14 Python
浏览器常用基本操作之python3+selenium4自动化测试(基础篇3)
May 21 Python
解决Jupyter notebook更换主题工具栏被隐藏及添加目录生成插件问题
Apr 20 #Python
jupyter 中文乱码设置编码格式 避免控制台输出的解决
Apr 20 #Python
Tensorflow tf.nn.atrous_conv2d如何实现空洞卷积的
Apr 20 #Python
python argparse模块通过后台传递参数实例
Apr 20 #Python
TensorFlow tf.nn.conv2d_transpose是怎样实现反卷积的
Apr 20 #Python
Tensorflow tf.nn.depthwise_conv2d如何实现深度卷积的
Apr 20 #Python
解决python脚本中error: unrecognized arguments: True错误
Apr 20 #Python
You might like
PHP中nowdoc和heredoc使用需要注意的一点
2014/03/21 PHP
php随机生成数字字母组合的方法
2015/03/18 PHP
php编写批量生成不重复的卡号密码代码
2015/05/14 PHP
Thinkphp批量更新数据的方法汇总
2016/06/29 PHP
tp5.1 框架数据库高级查询技巧实例总结
2020/05/25 PHP
网页图片延时加载的js代码
2010/04/22 Javascript
javascript下对于事件、事件流、事件触发的顺序随便说说
2010/07/17 Javascript
jQuery仿Excel表格编辑功能的实现代码
2013/05/01 Javascript
JS常见问题整理(持续更新)
2013/08/06 Javascript
JavaScript中的原型和继承详解(图文)
2014/07/18 Javascript
AngularJS基础 ng-mousemove 指令简单示例
2016/08/02 Javascript
把json格式的字符串转换成javascript对象或数组的方法总结
2016/11/03 Javascript
详解vue数组遍历方法forEach和map的原理解析和实际应用
2018/11/15 Javascript
js中实例与对象的区别讲解
2019/01/21 Javascript
JavaScript前端实现压缩图片功能
2020/03/06 Javascript
jQuery实现开关灯效果
2020/08/02 jQuery
python skimage 连通性区域检测方法
2018/06/21 Python
pytorch permute维度转换方法
2018/12/14 Python
python实现加密的方式总结
2020/01/19 Python
Python爬虫教程之利用正则表达式匹配网页内容
2020/12/08 Python
在python中对于bool布尔值的取反操作
2020/12/11 Python
HTML5中的postMessage API基本使用教程
2016/05/20 HTML / CSS
Columbia美国官网:美国著名的户外服装品牌
2016/11/24 全球购物
Omio西班牙:全欧洲低价大巴、火车和航班搜索和比价
2017/02/11 全球购物
Zavvi美国:英国娱乐之家
2017/03/19 全球购物
js实现弹框效果
2021/03/24 Javascript
网上开店必备创业计划书
2014/01/26 职场文书
菜篮子工程实施方案
2014/03/08 职场文书
结婚喜宴主持词
2014/03/14 职场文书
护理专业求职信
2014/06/15 职场文书
开展党的群众路线教育实践活动领导班子对照检查材料
2014/09/25 职场文书
行政文员岗位职责
2015/02/04 职场文书
财务总监岗位职责范本
2015/04/03 职场文书
三八节祝酒词
2015/08/11 职场文书
Python list去重且保持原顺序不变的方法
2021/04/03 Python
如何使用Maxwell实时同步mysql数据
2021/04/08 MySQL