快捷導(dǎo)航

jupyter notebook tensorflow打印device信息實(shí)例

更新時(shí)間：2020年04月20日 15:50:24 作者：li_sh1000

這篇文章主要介紹了jupyter notebook tensorflow打印device信息實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

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

補(bǔ)充知識(shí)：Jupyter無法在控制臺(tái)打印

因?yàn)閿?shù)據(jù)有中文，所以我特意在jupter前面設(shè)置了

reload(sys)
sys.setdefaultencoding("utf-8")

結(jié)果使用print語句的時(shí)候無法輸入內(nèi)容。究其原因，是因?yàn)閞eload的時(shí)候把sdout變?yōu)閕python的對(duì)象了，所以要臨時(shí)儲(chǔ)存一下stdout的對(duì)象。

不妨試一試以下代碼

import sys
stdo = sys.stdout
reload(sys)
sys.setdefaultencoding('utf-8')
sys.stdout= stdo

以上這篇jupyter notebook tensorflow打印device信息實(shí)例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

python每次處理固定個(gè)數(shù)的字符的方法總結(jié)
使用python每次處理固定個(gè)數(shù)的字符，很多情況下都會(huì)遇到。本文對(duì)可能的方法做下總結(jié)，供各位朋友學(xué)習(xí)參考
2013-01-01
Django unittest 設(shè)置跳過某些case的方法
今天小編就為大家分享一篇Django unittest 設(shè)置跳過某些case的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2018-12-12
深入淺析Pytorch中stack()方法
這篇文章主要介紹了Pytorch中stack()方法的總結(jié)及理解,本文通過實(shí)例圖文相結(jié)合給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2021-11-11
python順序執(zhí)行多個(gè)py文件的方法
今天小編大家分享一篇python順序執(zhí)行多個(gè)py文件的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2019-06-06
Python閉包和裝飾器用法實(shí)例詳解
這篇文章主要介紹了Python閉包和裝飾器用法,結(jié)合實(shí)例形式詳細(xì)分析了Python閉包和裝飾器的相關(guān)概念、原理、使用技巧與相關(guān)操作注意事項(xiàng),需要的朋友可以參考下
2019-05-05
pandas修改DataFrame列名的方法
下面小編就為大家分享一篇pandas修改DataFrame列名的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2018-04-04
Python中__slots__屬性介紹與基本使用方法
在Python中，每個(gè)類都有實(shí)例屬性。默認(rèn)情況下Python用一個(gè)字典來保存一個(gè)對(duì)象的實(shí)例屬性。這非常有用，因?yàn)樗试S我們?cè)谶\(yùn)行時(shí)去設(shè)置任意的新屬性。下面這篇文章主要給大家介紹了關(guān)于Python中__slots__屬性與基本使用方法的相關(guān)資料，需要的朋友可以參考下
2018-09-09
OpenCV+python實(shí)現(xiàn)膨脹和腐蝕的示例
這篇文章主要介紹了OpenCV+python實(shí)現(xiàn)膨脹和腐蝕的示例,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2020-12-12
pycharm中選中一個(gè)單詞替換所有重復(fù)單詞的實(shí)現(xiàn)方法
這篇文章主要介紹了pycharm中選中一個(gè)單詞替換所有重復(fù)單詞的實(shí)現(xiàn)方法，類似于sublime 里的ctrl+D功能，本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友參考下吧
2020-11-11
搭建pypi私有倉庫實(shí)現(xiàn)過程詳解
這篇文章主要介紹了搭建pypi私有倉庫實(shí)現(xiàn)過程詳解,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2020-11-11