tensorflow 使用ConfigProto&GPU
来源:互联网 发布:手机淘宝怎样装修店铺 编辑:程序博客网 时间:2024/06/05 08:14
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
- 1. tensorflow学习之ConfigProto&GPU
- tensorflow学习笔记(二十五):ConfigProto&GPU
- (转载)tensorflow学习笔记(二十五):ConfigProto&GPU
- [tensorflow]tensorflow ConfigProto
- tensorflow学习笔记:ConfigProto&GPU(转自http://blog.csdn.net/u012436149/article/details/53837651)
- Tensorflow ConfigProto & inter_/intra_op_parallelism_threads 整理
- tensorflow gpu使用情况
- tensorflow:GPU使用
- tensorflow GPU使用问题
- tensorflow使用gpu
- 【tensorflow 学习】 gpu使用
- tensorflow 不使用GPU 无法利用GPU
- Tensorflow指定使用哪个GPU
- 指定TensorFlow使用哪一个GPU
- 17、TensorFLow GPU 的使用
- sqlserver创建索引
- Nodejs ORM框架Sequelize快速入门
- 解决MySQL查询不区分大小写
- SSM整合---使用Maven依赖管理&构建
- mysql 字段添加唯一性约束
- tensorflow 使用ConfigProto&GPU
- js在循环中,传入i报错的问题
- jquery固定表头
- OpenLayers3三种动态效果
- 彩色宝石项链
- 搭建hexo博客(3)
- windows下 Mingw (gcc)编译openssl
- IntelliJ 使用 MyBatis Generator 详解
- 关于MYSQL group by 分组按时间取最大值的实现方法