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中表示字符串的三种方法
Sep 06 Python
Python数据处理numpy.median的实例讲解
Apr 02 Python
Python基于dom操作xml数据的方法示例
May 12 Python
python模块smtplib学习
May 22 Python
TensorFlow实现iris数据集线性回归
Sep 07 Python
tensorflow实现加载mnist数据集
Sep 08 Python
python numpy元素的区间查找方法
Nov 14 Python
python截取两个单词之间的内容方法
Dec 25 Python
运用Python的webbrowser实现定时打开特定网页
Feb 21 Python
python实现中文文本分句的例子
Jul 15 Python
Django项目后台不挂断运行的方法
Aug 31 Python
python 实现批量图片识别并翻译
Nov 02 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连mysql和oracle数据库性能比较
2006/10/09 PHP
Laravel5.1 框架响应基本用法实例分析
2020/01/04 PHP
JS JSON对象转为字符串的简单实现方法
2013/11/18 Javascript
jQuery打印图片pdf、txt示例代码
2014/07/22 Javascript
JavaScript使用addEventListener添加事件监听用法实例
2015/06/01 Javascript
javascript作用域问题实例分析
2015/07/13 Javascript
Node.js实用代码段之正确拼接Buffer
2016/03/17 Javascript
Javascript删除指定元素节点的方法
2016/06/21 Javascript
JS判断浏览器是否安装flash插件的简单方法
2016/09/13 Javascript
javascript 利用arguments实现可变长参数
2016/11/21 Javascript
任意Json转成无序列表的方法示例
2016/12/09 Javascript
深入学习nodejs中的async模块的使用方法
2017/07/12 NodeJs
Angularjs渲染的 using 指令的星级评分系统示例
2017/11/09 Javascript
vue安装和使用scss及sass与scss的区别详解
2018/10/15 Javascript
使用Angular自定义字段校验指令的方法示例
2019/02/01 Javascript
JS实现的杨辉三角【帕斯卡三角形】算法示例
2019/02/26 Javascript
ES6知识点整理之对象解构赋值应用示例
2019/04/17 Javascript
基于Vant UI框架实现时间段选择器
2020/12/24 Javascript
python生成器generator用法实例分析
2015/06/04 Python
Python读写unicode文件的方法
2015/07/10 Python
Python自定义主从分布式架构实例分析
2016/09/19 Python
python 接口返回的json字符串实例
2018/03/27 Python
使用python进行文本预处理和提取特征的实例
2018/06/05 Python
python 找出list中最大或者最小几个数的索引方法
2018/10/30 Python
python集成开发环境配置(pycharm)
2020/02/14 Python
Matplotlib 绘制饼图解决文字重叠的方法
2020/07/24 Python
python如何支持并发方法详解
2020/07/25 Python
英国轻奢珠宝品牌:Astley Clarke
2016/12/18 全球购物
英国在线定制百叶窗网站:Swift Direct Blinds
2020/02/25 全球购物
简历中求职的个人自我评价
2013/12/03 职场文书
银行开业庆典方案
2014/02/06 职场文书
人民调解员先进事迹材料
2014/05/08 职场文书
暑假学习心得体会
2014/09/02 职场文书
2014年连锁店圣诞节活动方案
2014/12/09 职场文书
工程合作意向书范本
2015/05/09 职场文书
2019广播稿怎么写
2019/04/17 职场文书