[tensorflow]tensorflow ConfigProto
来源:互联网 发布:linux修改ip配置文件 编辑:程序博客网 时间:2024/06/01 10:33
tensorflow ConfigProto
tf.ConfigProto一般用在创建session的时候。用来对session进行参数配置
with tf.Session(config = tf.ConfigProto(...),...)#tf.ConfigProto()的参数log_device_placement=True : 是否打印设备分配日志allow_soft_placement=True : 如果你指定的设备不存在,允许TF自动分配设备tf.ConfigProto(log_device_placement=True,allow_soft_placement=True)
控制GPU资源使用率
#allow growthconfig = tf.ConfigProto()config.gpu_options.allow_growth = Truesession = tf.Session(config=config, ...)# 使用allow_growth option,刚一开始分配少量的GPU容量,然后按需慢慢的增加,由于不会释放#内存,所以会导致碎片# per_process_gpu_memory_fractiongpu_options=tf.GPUOptions(per_process_gpu_memory_fraction=0.7)config=tf.ConfigProto(gpu_options=gpu_options)session = tf.Session(config=config, ...)#设置每个GPU应该拿出多少容量给进程使用,0.4代表 40%
控制使用哪块GPU
~/ CUDA_VISIBLE_DEVICES=0 python your.py#使用GPU0~/ CUDA_VISIBLE_DEVICES=0,1 python your.py#使用GPU0,1#注意单词不要打错
阅读全文
0 0
- [tensorflow]tensorflow ConfigProto
- tensorflow:ConfigProto&GPU
- tensorflow 使用ConfigProto&GPU
- TensorFlow--ConfigProto&GPU
- Tensorflow ConfigProto & inter_/intra_op_parallelism_threads 整理
- 1. tensorflow学习之ConfigProto&GPU
- tensorflow学习笔记(二十五):ConfigProto&GPU
- (转载)tensorflow学习笔记(二十五):ConfigProto&GPU
- tensorflow学习笔记:ConfigProto&GPU(转自http://blog.csdn.net/u012436149/article/details/53837651)
- tensorflow
- TensorFlow
- TensorFlow
- tensorflow
- tensorflow
- tensorflow
- Tensorflow
- Tensorflow
- tensorFlow
- POJ 2559 Largest Rectangle in a Histogram(单调栈)
- iOS之《Effective Objective-C 2.0》读书笔记(48)
- Spark 集群搭建从零开始之2 Spark单机伪分布式安装与配置
- phpcmsv9调用友情链接
- android GPS API
- [tensorflow]tensorflow ConfigProto
- 使用 jquery 获取一组或者单个 checkbox 的选中状态的值
- NAT详解
- 设计模式六大原则(3):依赖倒置原则
- 关于安卓glide图片框架加载的图片转换成圆形圆角的思路
- R语言 逐步回归分析 AIC
- 数据挖掘理论与算法
- SpringMVC入门学习
- Elasticsearch 5.6.1 发布,分布式搜索引擎