tensorflow学习笔记(二十五):ConfigProto&GPU
来源:互联网 发布:2016nba总决赛数据 编辑:程序博客网 时间:2024/05/27 14:14
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学习笔记(二十五):ConfigProto&GPU
- (转载)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--ConfigProto&GPU
- tensorflow学习笔记(二):windows10_64 tensorflow-gpu cuda
- 学习笔记(二十五)
- 深度学习笔记(二):基于tensorflow gpu版本的深度神经网络程序总览
- TensorFlow学习笔记(二十五)CNN的9大模型
- [tensorflow]tensorflow ConfigProto
- TensorFlow学习笔记(二):TensorFlow入门
- TensorFlow学习笔记(二)
- TensorFlow学习笔记(二)
- TensorFlow学习笔记(二)
- tensorflow学习笔记(三十一):构建多GPU代码
- Tensorflow实战学习(二十五)【自编码器】
- 我的caffe学习之路<第十七天>
- Shiro的标签应用
- 《从零开始学Swift》学习笔记(Day54)——抛出错误
- 十张数据图回顾雾霾,北京污染从南向北加深趋势明显
- javaFX - FileChooser
- tensorflow学习笔记(二十五):ConfigProto&GPU
- iOS命令行自动打包(archive)
- ssh免密码登录配置方法,(图示加命令)
- HTC vive基于unity的凝视交互功能(带HTC 插件)
- [译]Android Activity 和 Fragment 状态保存与恢复的最佳实践
- spark streaming 2 ParallelCollectionRDD
- Dos命令查看端口占用及关闭进程
- Gradle for Android-建立持续集成
- 配置NGINX同时运行 https 和 http