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中函数的参数定义和可变参数用法实例分析
Jun 04 Python
python 禁止函数修改列表的实现方法
Aug 03 Python
python 寻找list中最大元素对应的索引方法
Jun 28 Python
Python wxPython库使用wx.ListBox创建列表框示例
Sep 03 Python
django url到views参数传递的实例
Jul 19 Python
Python安装selenium包详细过程
Jul 23 Python
python ffmpeg任意提取视频帧的方法
Feb 21 Python
python 爬取小说并下载的示例
Dec 07 Python
Pytorch 图像变换函数集合小结
Feb 01 Python
Python实现文本文件拆分写入到多个文本文件的方法
Apr 18 Python
OpenCV-Python模板匹配人眼的实例
Jun 08 Python
浅谈哪个Python库才最适合做数据可视化
Jun 28 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扩展开发经验分享
2012/09/06 PHP
浅析php变量作用域的一些问题
2013/08/08 PHP
postfixadmin忘记密码后的修改密码方法详解
2016/07/20 PHP
laravel-admin 后台表格筛选设置默认的查询日期方法
2019/10/03 PHP
js压缩工具 yuicompressor 使用教程
2010/03/31 Javascript
JS 图片缩放效果代码
2010/06/09 Javascript
简单实用jquery版三级联动select示例
2013/07/04 Javascript
js中substring和substr的详细介绍与用法
2013/08/29 Javascript
jquery带下拉菜单和焦点图代码分享
2015/08/24 Javascript
JavaScript编写点击查看大图的页面半透明遮罩层效果实例
2016/05/09 Javascript
动态加载、移除js/css文件的示例代码
2018/03/20 Javascript
nodejs 日志模块winston的使用方法
2018/05/02 NodeJs
JS函数内部属性之arguments和this实例解析
2018/10/07 Javascript
Bootstarp在pycharm中的安装及简单的使用方法
2019/04/19 Javascript
详解基于Vue的支持数据双向绑定的select组件
2019/09/02 Javascript
解决小程序无法触发SESSION问题
2020/02/03 Javascript
vue-video-player 解决微信自动全屏播放问题(横竖屏导致样式错乱问题)
2020/02/25 Javascript
Python的Flask框架与数据库连接的教程
2015/04/20 Python
Python多进程库multiprocessing中进程池Pool类的使用详解
2017/11/24 Python
如何用Python实现简单的Markdown转换器
2018/07/16 Python
对python过滤器和lambda函数的用法详解
2019/01/21 Python
详解爬虫被封的问题
2019/04/23 Python
Python Django框架防御CSRF攻击的方法分析
2019/10/18 Python
python上传时包含boundary时的解决方法
2020/04/08 Python
python实现学生成绩测评系统
2020/06/22 Python
Python同时处理多个异常的方法
2020/07/28 Python
python实现图片转换成素描和漫画格式
2020/08/19 Python
Pycharm常用快捷键总结及配置方法
2020/11/14 Python
用CSS3绘制三角形的简单方法
2015/07/17 HTML / CSS
AmazeUI中各种的导航式菜单与解决方法
2020/08/19 HTML / CSS
英国最大的在线亚洲杂货店:Red Rickshaw
2020/03/22 全球购物
销售总经理岗位职责
2014/03/15 职场文书
小学教师自我评价
2015/03/04 职场文书
工作简历的自我评价
2019/05/16 职场文书
教你如何使用Python下载B站视频的详细教程
2021/04/29 Python
MySQL修炼之联结与集合浅析
2021/10/05 MySQL