tensorflow:ConfigProto&GPU
来源:互联网 发布:网络犯罪调查分集剧情 编辑:程序博客网 时间:2024/06/18 02:50
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使用说明
- 安装windows tensorflow-gpu
- Install GPU Tensorflow
- Tensorflow gpu 安装
- windows gpu tensorflow anaconda
- TensorFlow(GPU) 安装
- win10+ubutun+tensorflow+gpu
- TensorFlow GPU版安装
- C条件编译在软件开发中的运用
- 如何去设计一个自适应的网页设计或HTMl5
- 笛卡尔树
- 以注解的形式来实现一个springmvc的简单范例
- 21. Merge Two Sorted Lists
- tensorflow:ConfigProto&GPU
- Kafka中时间轮分析与Java实现
- 方便记忆的电话号码
- 区块链2.0(四):智能资产构建去中心化的资产管理系统
- CCF 201604-2 俄罗斯方块
- 调试技巧——宏定义开关和printf
- 利用Python+OpenCV与GDAL批量裁剪图像
- 个人项目总结---微信扫描二维码强制群发
- php小事日常记