(转载)tensorflow学习笔记(二十五):ConfigProto&GPU
2017-09-26 11:40
555 查看
tensorflow ConfigProto
tf.ConfigProto一般用在创建
session的时候。用来对
session进行参数配置
with tf.Session(config = tf.ConfigProto(...),...)1
[/code]
#tf.ConfigProto()的参数 log_device_placement=True : 是否打印设备分配日志 allow_soft_placement=True : 如果你指定的设备不存在,允许TF自动分配设备 tf.ConfigProto(log_device_placement=True,allow_soft_placement=True)1
2
3
4
[/code]
控制GPU资源使用率
#allow growth config = tf.ConfigProto() config.gpu_options.allow_growth = True session = tf.Session(config=config, ...) # 使用allow_growth option,刚一开始分配少量的GPU容量,然后按需慢慢的增加,由于不会释放 #内存,所以会导致碎片1
2
3
4
5
6
[/code]
# per_process_gpu_memory_fraction gpu_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%1
2
3
4
5
[/code]
控制使用哪块GPU
~/ CUDA_VISIBLE_DEVICES=0 python your.py#使用GPU0 ~/ CUDA_VISIBLE_DEVICES=0,1 python your.py#使用GPU0,1 #注意单词不要打错1
2
3
[/code]
相关文章推荐
- tensorflow学习笔记(二十五):ConfigProto&GPU
- tensorflow学习笔记:ConfigProto&GPU(转自http://blog.csdn.net/u012436149/article/details/53837651)
- 1. tensorflow学习之ConfigProto&GPU
- tensorflow 使用ConfigProto&GPU
- tensorflow:ConfigProto&GPU
- tensorflow中使用tf.ConfigProto()配置Session运行参数&&GPU设备指定
- tensorflow:ConfigProto&GPU
- ConfigProto&GPU
- TensorFlow--ConfigProto&GPU
- Tensorflow ConfigProto & inter_/intra_op_parallelism_threads 整理
- cocos2d-x, protobuf, no config.h, #error "No suitable threading library available."
- <转载>Js中Prototype、__proto__、Constructor、Object、Function关系介绍
- There is a cycle in the hierarchy & json-lib 之jsonConfig详细使用(转载写的不错)
- (转载) MTK申请内存
- android4.0 禁止横竖屏切换使用 android:configChanges="orientation|keyboardHidden"无效
- Spring 开启Annotation <context:annotation-config> 和 <context:component-scan>诠释及区别
- 遇到的Linux反汇编问题 & 转载一篇好文章
- 关于android:configChanges="keyboardHidden|orientation"
- <转载>Android 通过蓝牙控制小车源代码+视频
- D3D9 GPU Hacks (转载)