快捷導(dǎo)航

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

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

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

juypter notebook中直接使用log_device_placement=True打印不出來(lái)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來(lái)輸出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無(wú)法在控制臺(tái)打印

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

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

結(jié)果使用print語(yǔ)句的時(shí)候無(wú)法輸入內(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í)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

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