jupyter notebook tensorflow打印device信息实例
更新时间：2020年04月20日 15:50:24 作者：li_sh1000
这篇文章主要介绍了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信息实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要为大家介绍了python利用opencv调用摄像头实现目标检测的示例代码，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-05-05
大家好，本篇文章主要讲的是用Python实现zip密码破解实例，感兴趣的同学赶快来看一看吧，对你有帮助的话记得收藏一下2022-01-01
最近上网的时候看到了一个有趣的东西,叫做人物动漫化,尝试着用python实现了,所以下面这篇文章主要给大家介绍了关于Python调用腾讯API进行人像动漫化效果的相关资料,需要的朋友可以参考下2023-06-06
成语填空想必大家都是十分熟悉的了，特别是有在上小学的家长肯定都有十分深刻的印象。当然了你也别小看了成语调控小游戏，有的时候知识储备不够，你还真的不一定猜得出来是什么。本文就来用Python编写一个简单的成语填空游戏，感兴趣的可以了解下2023-02-02
Python协程操作之gevent(yield阻塞，greenlet)，协程实现多任务(有规律的交替协作执行)用法详解
这篇文章主要介绍了Python协程操作之gevent(yield阻塞，greenlet)，协程实现多任务(有规律的交替协作执行)用法,结合实例形式较为详细的分析了协程的功能、原理及gevent、greenlet实现协程,以及协程实现多任务相关操作技巧,需要的朋友可以参考下2019-10-10
本文主要介绍了django channels使用和配置及实现群聊，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-05-05
今天小编就为大家分享一篇python实现控制台打印的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
这篇文章主要介绍了Python使用sorted对字典的key或value排序，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-11-11
Beautiful Soup是python的一个库，最主要的功能是从网页抓取数据。今天通过本文给大家分享python beautiful soup库入门教程，需要的朋友参考下吧2021-08-08
这篇文章主要介绍了使用 OpenCV-Python 识别答题卡判卷,本文分步骤通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2021-12-12
最新评论