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無法在控制臺打印
因為數(shù)據(jù)有中文，所以我特意在jupter前面設置了
reload(sys)
sys.setdefaultencoding("utf-8")
結(jié)果使用print語句的時候無法輸入內(nèi)容。究其原因，是因為reload的時候把sdout變?yōu)閕python的對象了，所以要臨時儲存一下stdout的對象。
不妨試一試以下代碼
import sys
stdo = sys.stdout
reload(sys)
sys.setdefaultencoding('utf-8')
sys.stdout= stdo
以上這篇jupyter notebook tensorflow打印device信息實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
您可能感興趣的文章:
相關文章
python每次處理固定個數(shù)的字符的方法總結(jié)
使用python每次處理固定個數(shù)的字符，很多情況下都會遇到。本文對可能的方法做下總結(jié)，供各位朋友學習參考2013-01-01
今天小編就為大家分享一篇Django unittest 設置跳過某些case的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
這篇文章主要介紹了Pytorch中stack()方法的總結(jié)及理解,本文通過實例圖文相結(jié)合給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-11-11
今天小編大家分享一篇python順序執(zhí)行多個py文件的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
這篇文章主要介紹了Python閉包和裝飾器用法,結(jié)合實例形式詳細分析了Python閉包和裝飾器的相關概念、原理、使用技巧與相關操作注意事項,需要的朋友可以參考下2019-05-05
下面小編就為大家分享一篇pandas修改DataFrame列名的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
在Python中，每個類都有實例屬性。默認情況下Python用一個字典來保存一個對象的實例屬性。這非常有用，因為它允許我們在運行時去設置任意的新屬性。下面這篇文章主要給大家介紹了關于Python中__slots__屬性與基本使用方法的相關資料，需要的朋友可以參考下2018-09-09
這篇文章主要介紹了OpenCV+python實現(xiàn)膨脹和腐蝕的示例,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-12-12
pycharm中選中一個單詞替換所有重復單詞的實現(xiàn)方法
這篇文章主要介紹了pycharm中選中一個單詞替換所有重復單詞的實現(xiàn)方法，類似于sublime 里的ctrl+D功能，本文通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友參考下吧2020-11-11
這篇文章主要介紹了搭建pypi私有倉庫實現(xiàn)過程詳解,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下2020-11-11