快捷導(dǎo)航

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前面設(shè)置了

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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

python利用opencv調(diào)用攝像頭實現(xiàn)目標檢測
這篇文章主要為大家介紹了python利用opencv調(diào)用攝像頭實現(xiàn)目標檢測的示例代碼，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2022-05-05
用Python實現(xiàn)zip密碼破解實例
大家好，本篇文章主要講的是用Python實現(xiàn)zip密碼破解實例，感興趣的同學趕快來看一看吧，對你有幫助的話記得收藏一下
2022-01-01
Python調(diào)用騰訊API進行人像動漫化效果實例
最近上網(wǎng)的時候看到了一個有趣的東西,叫做人物動漫化,嘗試著用python實現(xiàn)了,所以下面這篇文章主要給大家介紹了關(guān)于Python調(diào)用騰訊API進行人像動漫化效果的相關(guān)資料,需要的朋友可以參考下
2023-06-06
基于Python實現(xiàn)成語填空游戲的示例代碼
成語填空想必大家都是十分熟悉的了，特別是有在上小學的家長肯定都有十分深刻的印象。當然了你也別小看了成語調(diào)控小游戲，有的時候知識儲備不夠，你還真的不一定猜得出來是什么。本文就來用Python編寫一個簡單的成語填空游戲，感興趣的可以了解下
2023-02-02
Python協(xié)程操作之gevent(yield阻塞，greenlet)，協(xié)程實現(xiàn)多任務(wù)(有規(guī)律的交替協(xié)作執(zhí)行)用法詳解
這篇文章主要介紹了Python協(xié)程操作之gevent(yield阻塞，greenlet)，協(xié)程實現(xiàn)多任務(wù)(有規(guī)律的交替協(xié)作執(zhí)行)用法,結(jié)合實例形式較為詳細的分析了協(xié)程的功能、原理及gevent、greenlet實現(xiàn)協(xié)程,以及協(xié)程實現(xiàn)多任務(wù)相關(guān)操作技巧,需要的朋友可以參考下
2019-10-10
django channels使用和配置及實現(xiàn)群聊
本文主要介紹了django channels使用和配置及實現(xiàn)群聊，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2022-05-05
python實現(xiàn)控制臺打印的方法
今天小編就為大家分享一篇python實現(xiàn)控制臺打印的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-01-01
Python使用sorted對字典的key或value排序
這篇文章主要介紹了Python使用sorted對字典的key或value排序，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-11-11
python beautiful soup庫入門安裝教程
Beautiful Soup是python的一個庫，最主要的功能是從網(wǎng)頁抓取數(shù)據(jù)。今天通過本文給大家分享python beautiful soup庫入門教程，需要的朋友參考下吧
2021-08-08
使用?OpenCV-Python?識別答題卡判卷功能
這篇文章主要介紹了使用?OpenCV-Python?識別答題卡判卷,本文分步驟通過實例代碼給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2021-12-12