脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → 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信息实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

相关文章

  • python每次处理固定个数的字符的方法总结

    python每次处理固定个数的字符的方法总结

    使用python每次处理固定个数的字符，很多情况下都会遇到。本文对可能的方法做下总结，供各位朋友学习参考
    2013-01-01
  • Django unittest 设置跳过某些case的方法

    Django unittest 设置跳过某些case的方法

    今天小编就为大家分享一篇Django unittest 设置跳过某些case的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2018-12-12
  • 深入浅析Pytorch中stack()方法

    深入浅析Pytorch中stack()方法

    这篇文章主要介绍了Pytorch中stack()方法的总结及理解,本文通过实例图文相结合给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
    2021-11-11
  • python顺序执行多个py文件的方法

    python顺序执行多个py文件的方法

    今天小编大家分享一篇python顺序执行多个py文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2019-06-06
  • Python闭包和装饰器用法实例详解

    Python闭包和装饰器用法实例详解

    这篇文章主要介绍了Python闭包和装饰器用法,结合实例形式详细分析了Python闭包和装饰器的相关概念、原理、使用技巧与相关操作注意事项,需要的朋友可以参考下
    2019-05-05
  • pandas修改DataFrame列名的方法

    pandas修改DataFrame列名的方法

    下面小编就为大家分享一篇pandas修改DataFrame列名的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2018-04-04
  • Python中__slots__属性介绍与基本使用方法

    Python中__slots__属性介绍与基本使用方法

    在Python中，每个类都有实例属性。默认情况下Python用一个字典来保存一个对象的实例属性。这非常有用，因为它允许我们在运行时去设置任意的新属性。下面这篇文章主要给大家介绍了关于Python中__slots__属性与基本使用方法的相关资料，需要的朋友可以参考下
    2018-09-09
  • OpenCV+python实现膨胀和腐蚀的示例

    OpenCV+python实现膨胀和腐蚀的示例

    这篇文章主要介绍了OpenCV+python实现膨胀和腐蚀的示例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • pycharm中选中一个单词替换所有重复单词的实现方法

    pycharm中选中一个单词替换所有重复单词的实现方法

    这篇文章主要介绍了pycharm中选中一个单词替换所有重复单词的实现方法，类似于sublime 里的ctrl+D功能，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友参考下吧
    2020-11-11
  • 搭建pypi私有仓库实现过程详解

    搭建pypi私有仓库实现过程详解

    这篇文章主要介绍了搭建pypi私有仓库实现过程详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-11-11

最新评论