TensorFlow中tf.ConfigProto()配置Sesion运算方式

博主个人网站：https://chenzhen.online

tf.configProto用于在创建Session的时候配置Session的运算方式，即使用GPU运算或CPU运算；

1. `tf.ConfigProto()`中的基本参数：

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

参数	用法
`log_device_placement=True`	设置为`True`时，会打印出TensoFlow使用了哪种操作；
`allow_soft_placement=True`	当运行设备不满足要求时，会自动分配GPU或CPU；
`inter_op_parallelism_threads=0`	一个操作内部并行运算的线程数；
`intra_op_parallelism_threads=0`	多个操作并行运算的线程数；

2. `tf.ConfigProto`配置GPU

2.1 判断TensorFlow是否能够使用GPU运算

result = tf.test.is_built_with_cuda()
print(result)

2.2 两种方式配置使用具体哪块GPU

# 方式一：在python程序中设置
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
os.environ['CUDA_VISIBLE_DEVICES'] = '0, 1'# 方式二：在执行.py文件时，指定具体GPU块
CUDA_VISIBLE_DEVICES=0,1 python test.py

2.3 动态申请GPU显存

session_config = tf.ConfigProto()  # 配置见第1节
session_config.gpu_options.allow_growth = True  # 动态申请
sess = tf.Session(config=session_config)

2.4 限制GPU的使用率

session_configsession_config = tf.ConfigProto()  # 配置见第1节
session_config.gpu_options.per_process_gpu_memory_fraction = 0.4  # 占用40%显存
session = tf.Session(config=session_config)

Reference

https://blog.csdn.net/qq_31261509/article/details/79746114
https://blog.csdn.net/dcrmg/article/details/79091941

转载于:https://www.cnblogs.com/chenzhen0530/p/10623856.html

TensorFlow中tf.ConfigProto()配置Sesion运算方式相关推荐

Tensorflow中tf.ConfigProto()详解
参考Tensorflow Machine Leanrning Cookbook tf.ConfigProto()主要的作用是配置tf.Session的运算方式,比如gpu运算或者cpu运算具体代码如 ...
tensorflow中使用tf.ConfigProto()配置Session运行参数GPU设备指定
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明. 本文链接:https://blog.csdn.net/dcrmg/article/detail ...
TensorFlow中RNN实现的正确打开方式
上周写的文章<完全图解RNN.RNN变体.Seq2Seq.Attention机制>介绍了一下RNN的几种结构,今天就来聊一聊如何在TensorFlow中实现这些结构,这篇文章的主要内容为: ...
TensorFlow中RNN实现的正确打开方式（转）
上周写的文章<完全图解RNN.RNN变体.Seq2Seq.Attention机制>介绍了一下RNN的几种结构,今天就来聊一聊如何在TensorFlow中实现这些结构,这篇文章的主要内容为: ...
tensorflow中tf.gradients()解析
tf.gradients()解析及grad_ys在xs为(?, 1)时的理解问题简介使用tensorflow1.15学习时,有一项tf.gradients的代码,其中用到了grad_ys这个参数, ...
python中logging(日志)配置三种方式
超详细日志文档 python中,logging由logger.handler.filter.formater四个部分组成: logger(记录器):提供我们记录日志的方法: handler(处理器): ...
tensorflow中tf.nn.xw_plus_b
tf.nn.xw_plus_b((x, weights) + biases) 相当于tf.matmul(x, weights) + biases #-*-coding:utf8-*- import t ...
TensorFlow中 tf.space_to_depth()函数的用法
目录一.函数定义二.解释范例三.代码验证一.函数定义通俗易懂些,就是把输入为[batch, height, width, channels]形式的Tensor,其在height和width维 ...
Spring中三种配置Bean的方式
1.使用xml配置文件由Spring Framework本身决定,此处不详细说. 2.使用Java注解的配置如果一个类使用了@Service,那么此类将自动注册成一个bean,不需要再在appli ...
TensorFlow 中 tf.app.flags.FLAGS 的用法介绍
转载自:https://blog.csdn.net/lyc_yongcai/article/details/73456960 下面介绍 tf.app.flags.FLAGS 的使用,主要是在用命令行执 ...

TensorFlow中tf.ConfigProto()配置Sesion运算方式

1. `tf.ConfigProto()`中的基本参数：

2. `tf.ConfigProto`配置GPU

2.1 判断TensorFlow是否能够使用GPU运算

2.2 两种方式配置使用具体哪块GPU

2.3 动态申请GPU显存

2.4 限制GPU的使用率

Reference

TensorFlow中tf.ConfigProto()配置Sesion运算方式相关推荐

最新文章

热门文章

TensorFlow中tf.ConfigProto()配置Sesion运算方式

1. tf.ConfigProto()中的基本参数：

2. tf.ConfigProto配置GPU

2.1 判断TensorFlow是否能够使用GPU运算

2.2 两种方式配置使用具体哪块GPU

2.3 动态申请GPU显存

2.4 限制GPU的使用率

Reference

TensorFlow中tf.ConfigProto()配置Sesion运算方式相关推荐

最新文章

热门文章

1. `tf.ConfigProto()`中的基本参数：

2. `tf.ConfigProto`配置GPU