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非递归全排列实现方法
Apr 10 Python
Python单体模式的几种常见实现方法详解
Jul 28 Python
python实时监控cpu小工具
Jun 21 Python
Django 中使用流响应处理视频的方法
Jul 20 Python
解决python给列表里添加字典时被最后一个覆盖的问题
Jan 21 Python
python读取.mat文件的数据及实例代码
Jul 12 Python
python 求定积分和不定积分示例
Nov 20 Python
django数据模型(Model)的字段类型解析
Dec 25 Python
Python Selenium参数配置方法解析
Jan 19 Python
python中sympy库求常微分方程的用法
Apr 28 Python
Python文件操作模拟用户登陆代码实例
Jun 09 Python
python中pickle模块浅析
Dec 29 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设计模式之迭代器模式的深入解析
2013/06/13 PHP
php的array数组和使用实例简明教程(容易理解)
2014/03/20 PHP
php+ajax实现图片文件上传功能实例
2014/06/17 PHP
谈谈php对接芝麻信用踩的坑
2016/12/01 PHP
laravel配置Redis多个库的实现方法
2019/04/10 PHP
PHP信号处理机制的操作代码讲解
2019/04/19 PHP
Javascript基础教程之数据类型 (字符串 String)
2015/01/18 Javascript
javascript中使用未定义变量或值的情况分析
2016/07/19 Javascript
Bootstrap CSS布局之列表
2016/12/15 Javascript
Angular使用ng-messages与PHP进行表单数据验证
2016/12/28 Javascript
轻松理解JavaScript闭包
2017/03/14 Javascript
利用node.js如何搭建一个简易的即时响应服务器
2017/05/28 Javascript
node+express+ejs使用模版引擎做的一个示例demo
2017/09/18 Javascript
angularjs 动态从后台获取下拉框的值方法
2018/08/13 Javascript
详解如何配置vue-cli3.0的vue.config.js
2018/08/23 Javascript
JS基于Location实现访问Url、重定向及刷新页面的方法分析
2018/12/03 Javascript
vue中过滤器filter的讲解
2019/01/21 Javascript
详解vue-element Tree树形控件填坑路
2019/03/26 Javascript
javascript实现的时间格式加8小时功能示例
2019/06/13 Javascript
[03:52]显微镜下的DOTA2第三期——英雄在无聊的时候干什么
2014/06/20 DOTA
[03:42]2014DOTA2国际邀请赛 第三日比赛排位扑朔迷离
2014/07/12 DOTA
python制作花瓣网美女图片爬虫
2015/10/28 Python
Python中几种导入模块的方式总结
2017/04/27 Python
Python和Go语言的区别总结
2019/02/20 Python
python爬虫实现获取下一页代码
2020/03/13 Python
django rest framework serializer返回时间自动格式化方法
2020/03/31 Python
Python3爬虫关于代理池的维护详解
2020/07/30 Python
路易威登和香奈儿手袋:LuxeDH
2017/01/12 全球购物
泰国演唱会订票网站:StubHub泰国
2018/02/26 全球购物
Blancsom美国/加拿大:服装和生活用品供应商
2018/07/27 全球购物
Groupon比利时官方网站:特卖和网上购物高达-70%
2019/08/09 全球购物
LVMH旗下最大的奢侈品网站平台:24S
2020/05/24 全球购物
大学生表扬信范文
2014/01/09 职场文书
少先队入队活动方案
2014/02/08 职场文书
初中三年毕业生的自我评价分享
2014/02/14 职场文书
Python内置的数据类型及使用方法
2022/04/13 Python