Tensorflow中tf.ConfigProto()的用法详解


Posted in Python onFebruary 06, 2020

参考Tensorflow Machine Leanrning Cookbook

tf.ConfigProto()主要的作用是配置tf.Session的运算方式,比如gpu运算或者cpu运算

具体代码如下:

import tensorflow as tf

session_config = tf.ConfigProto(
   log_device_placement=True,
   inter_op_parallelism_threads=0,
   intra_op_parallelism_threads=0,
   allow_soft_placement=True)

sess = tf.Session(config=session_config)

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2,3], name='b')
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)
print(sess.run(c))

具体解释

log_device_placement=True

设置为True时,会打印出TensorFlow使用了那种操作

inter_op_parallelism_threads=0

设置线程一个操作内部并行运算的线程数,比如矩阵乘法,如果设置为0,则表示以最优的线程数处理

intra_op_parallelism_threads=0

设置多个操作并行运算的线程数,比如 c = a + b,d = e + f . 可以并行运算

allow_soft_placement=True

有时候,不同的设备,它的cpu和gpu是不同的,如果将这个选项设置成True,那么当运行设备不满足要求时,会自动分配GPU或者CPU。

其他选项

当使用GPU时候,Tensorflow运行自动慢慢达到最大GPU的内存

session_config.gpu_options.allow_growth = True

当使用GPU时,设置GPU内存使用最大比例

session_config.gpu_options.per_process_gpu_memory_fraction = 0.4

是否能够使用GPU进行运算

tf.test.is_built_with_cuda()

另外的处理方法

import tensorflow as tf

sess = tf.Session()

with tf.device('/cpu:0'):
  a = tf.constant([1.0, 3.0, 5.0], shape=[1, 3])
  b = tf.constant([2.0, 4.0, 6.0], shape=[3, 1])

  with tf.device('/gpu:0'):
    c = tf.matmul(a, b)
    c = tf.reshape(c, [-1])

  with tf.device('/gpu:0'):
    d = tf.matmul(b, a)
    flat_d = tf.reshape(d, [-1])

  combined = tf.multiply(c, flat_d)
  print(sess.run(combined))

以上这篇Tensorflow中tf.ConfigProto()的用法详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python使用装饰器和线程限制函数执行时间的方法
Apr 18 Python
基于Python实现对PDF文件的OCR识别
Aug 05 Python
Python基于回溯法子集树模板解决0-1背包问题实例
Sep 02 Python
Python基于pygame模块播放MP3的方法示例
Sep 30 Python
python实现求解列表中元素的排列和组合问题
Mar 15 Python
Python实现的对本地host127.0.0.1主机进行扫描端口功能示例
Feb 15 Python
python中PS 图像调整算法原理之亮度调整
Jun 28 Python
python图的深度优先和广度优先算法实例分析
Oct 26 Python
使用PyCharm官方中文语言包汉化PyCharm
Nov 18 Python
Python实现自动整理文件的脚本
Dec 17 Python
如何用python 操作zookeeper
Dec 28 Python
Django框架中模型的用法
Jun 10 Python
Python continue语句实例用法
Feb 06 #Python
Tensorflow限制CPU个数实例
Feb 06 #Python
如何解决tensorflow恢复模型的特定值时出错
Feb 06 #Python
python range实例用法分享
Feb 06 #Python
tensorflow 报错unitialized value的解决方法
Feb 06 #Python
TensorFlow实现保存训练模型为pd文件并恢复
Feb 06 #Python
解决TensorFlow模型恢复报错的问题
Feb 06 #Python
You might like
说明的比较细的php 正则学习实例
2008/07/30 PHP
CL vs ForZe BO5 第二场 2.13
2021/03/10 DOTA
用JavaScript调用WebService的示例
2008/04/07 Javascript
基于Jquery的仿照flash放大图片效果代码
2011/03/16 Javascript
php与js的区别是什么
2013/08/05 Javascript
jquery插件开发之实现jquery手风琴功能分享
2014/03/10 Javascript
JavaScript学习笔记之Function对象
2015/01/22 Javascript
js实现上传文件添加和删除文件选择框
2016/10/24 Javascript
JavaScript将base64图片转换成formData并通过AJAX提交的实现方法
2016/10/24 Javascript
JavaScript+Html5实现按钮复制文字到剪切板功能(手机网页兼容)
2017/03/30 Javascript
js canvas实现擦除效果示例代码
2017/04/26 Javascript
JavaScript中正则表达式使数字、中文或指定字符高亮显示
2017/10/31 Javascript
jquery中ajax请求后台数据成功后既不执行success也不执行error的完美解决方法
2017/12/24 jQuery
常用的 JS 排序算法 整理版
2018/04/05 Javascript
微信小程序实现的picker多级联动功能示例
2019/05/23 Javascript
ssm+vue前后端分离框架整合实现(附源码)
2020/07/08 Javascript
javascript实现放大镜功能
2020/12/09 Javascript
python中列表元素连接方法join用法实例
2015/04/07 Python
讲解Python的Scrapy爬虫框架使用代理进行采集的方法
2016/02/18 Python
Python中的迭代器与生成器高级用法解析
2016/06/28 Python
Django之无名分组和有名分组的实现
2019/04/16 Python
在django admin中添加自定义视图的例子
2019/07/26 Python
python实现大学人员管理系统
2019/10/25 Python
Pycharm中安装wordcloud等库失败问题及终端通过pip安装的Python库如何添加到Pycharm解释器中(推荐)
2020/05/10 Python
在keras 中获取张量 tensor 的维度大小实例
2020/06/10 Python
如何用 Python 处理不平衡数据集
2021/01/04 Python
单位创先争优活动方案
2014/01/26 职场文书
公司开业庆典主持词
2014/03/21 职场文书
2014年双拥工作总结
2014/11/21 职场文书
2014年化妆品销售工作总结
2014/12/01 职场文书
画展邀请函
2015/01/31 职场文书
写给女朋友的检讨书
2015/05/06 职场文书
穷人该怎么创业?谨记以下几点
2019/07/11 职场文书
MySQL里面的子查询的基本使用
2021/08/02 MySQL
MySQL 开窗函数
2022/02/15 MySQL
win10以太网连接不上怎么办?Win10连接以太网详细教程
2022/04/08 数码科技