没有合适的资源?快使用搜索试试~ 我知道了~
jupyter notebook tensorflow打印device信息实例
0 下载量 167 浏览量
2020-12-20
12:10:28
上传
评论
收藏 41KB PDF 举报
温馨提示
试读
1页
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 sessi
资源详情
资源评论
资源推荐
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信息实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希
望大家多多支持软件开发网。
您可能感兴趣的文章您可能感兴趣的文章:jupyter notebook中新建cell的方法与快捷键操作Jupyter Notebook折叠输出的内容实例浅谈
JupyterNotebook导出pdf解决中文的问题解决Jupyter notebook中.py与.ipynb文件的import问题jupyter notebook 添加kernel
permission denied的操作Jupyter Notebook的连接密码 token查询方式快速解决jupyter notebook启动需要密码的问题jupyter
notebook 的工作空间设置操作查看jupyter notebook每个单元格运行时间实例
weixin_38650629
- 粉丝: 4
- 资源: 897
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功
评论0