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中pygame模块用法实例
Oct 09 Python
python打开url并按指定块读取网页内容的方法
Apr 29 Python
web.py 十分钟创建简易博客实现代码
Apr 22 Python
Python实现购物车功能的方法分析
Nov 10 Python
让Python更加充分的使用Sqlite3
Dec 11 Python
Python从文件中读取指定的行以及在文件指定位置写入
Sep 06 Python
python中class的定义及使用教程
Sep 18 Python
django框架F&Q 聚合与分组操作示例
Dec 12 Python
.dcm格式文件软件读取及python处理详解
Jan 16 Python
Python 实现平台类游戏添加跳跃功能
Mar 27 Python
python脚本定时发送邮件
Dec 22 Python
python中validators库的使用方法详解
Sep 23 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的5个安全措施小结
2012/07/17 PHP
PHP加Nginx实现动态裁剪图片方案
2014/03/10 PHP
php实现三级级联下拉框
2016/04/17 PHP
ucenter中词语过滤原理分析
2016/07/13 PHP
超链接的禁用属性Disabled使用示例
2014/07/31 Javascript
微信小程序 教程之wxapp视图容器 swiper
2016/10/19 Javascript
JS实现给对象动态添加属性的方法
2017/01/05 Javascript
Angular 4.x 动态创建表单实例
2017/04/25 Javascript
JS 学习总结之正则表达式的懒惰性和贪婪性
2017/07/03 Javascript
vue2.0使用swiper组件实现轮播效果
2017/11/27 Javascript
jQuery判断自定义属性data-val用法示例
2019/01/07 jQuery
jquery+css实现Tab栏切换的代码实例
2019/05/14 jQuery
vue iview的菜单组件Mune 点击不高亮的解决方案
2019/11/01 Javascript
JS实现图片幻灯片效果代码实例
2020/05/21 Javascript
python检测远程端口是否打开的方法
2015/03/14 Python
Python3实现从文件中读取指定行的方法
2015/05/22 Python
python3实现公众号每日定时发送日报和图片
2018/02/24 Python
对python中的 os.mkdir和os.mkdirs详解
2018/10/16 Python
python实现简单flappy bird
2018/12/24 Python
Python 仅获取响应头, 不获取实体的实例
2019/08/21 Python
python解释器spython使用及原理解析
2019/08/24 Python
使用Python制作一个打字训练小工具
2019/10/01 Python
python paramiko远程服务器终端操作过程解析
2019/12/14 Python
Softmax函数原理及Python实现过程解析
2020/05/22 Python
Python基于yaml文件配置logging日志过程解析
2020/06/23 Python
CSS3制作酷炫的三维相册效果
2016/07/01 HTML / CSS
Lacoste澳大利亚官网:服装、鞋类及配饰
2018/11/14 全球购物
安德玛比利时官网:Under Armour比利时
2019/08/28 全球购物
const和static readonly区别
2013/05/20 面试题
幼儿园爱国卫生月活动总结
2014/06/30 职场文书
质量在我心中演讲稿
2014/09/02 职场文书
人事代理委托书
2014/09/27 职场文书
党的群众路线教育实践活动个人对照检查材料(公安)
2014/11/05 职场文书
2014年物资管理工作总结
2014/12/02 职场文书
中秋节寄语2015
2015/03/24 职场文书
2015年服务员工作总结
2015/04/08 职场文书