脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → jupyter notebook tensorflow device

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信息实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • pytorch 网络参数 weight bias 初始化详解

    pytorch 网络参数 weight bias 初始化详解

    这篇文章主要介绍了pytorch 网络参数 weight bias 初始化详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-06-06
  • MoviePy简介及Python视频剪辑自动化

    MoviePy简介及Python视频剪辑自动化

    MoviePy是一个用于视频编辑的Python模块，可用于基本操作（如剪切、拼接、字幕插入）、视频合成、视频处理或创建高级效果等。本文给大家介绍MoviePy简介及Python视频剪辑自动化的相关知识，感兴趣的朋友一起看看吧
    2020-12-12
  • Python time时间格式化操作指南

    Python time时间格式化操作指南

    这篇文章主要给大家介绍了关于Python time时间格式化操作的相关资料,Python中日期格式化是非常常见的操作,Python中能用很多方式处理日期和时间,转换日期格式是一个常见的功能,需要的朋友可以参考下
    2023-10-10
  • python实现经纬度采样的示例代码

    python实现经纬度采样的示例代码

    这篇文章主要介绍了python实现经纬度采样的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • Python中使用Queue和Condition进行线程同步的方法

    Python中使用Queue和Condition进行线程同步的方法

    这篇文章主要介绍了Python中使用Queue模块和Condition对象进行线程同步的方法,配合threading模块下的线程编程进行操作的实例,需要的朋友可以参考下
    2016-01-01
  • python 中不同包 类 方法 之间的调用详解

    python 中不同包 类 方法 之间的调用详解

    这篇文章主要介绍了python 中不同包 类 方法 之间的调用详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-03-03
  • Django REST Framework序列化外键获取外键的值方法

    Django REST Framework序列化外键获取外键的值方法

    今天小编就为大家分享一篇Django REST Framework序列化外键获取外键的值方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-07-07
  • Python中Array特性与应用实例深入探究

    Python中Array特性与应用实例深入探究

    这篇文章主要为大家介绍了Python中Array特性与应用实例深入探究,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2024-01-01
  • Python入门教程(三十一)Python的Try和Except

    Python入门教程(三十一)Python的Try和Except

    这篇文章主要介绍了Python入门教程(三十一)Python的Try Except,当我们调用Python并发生错误或异常时,通常会停止并生成错误消息,
    可以使用try语句处理这些异常,需要的朋友可以参考下
    2023-05-05
  • Python实现批量填补遥感影像的无效值NoData

    Python实现批量填补遥感影像的无效值NoData

    这篇文章主要为大家介绍了如何基于Python中ArcPy模块，对大量栅格遥感影像文件批量进行无效值（NoData值）填充的方法，感兴趣的小伙伴可以了解一下
    2023-06-06

最新评论