tensorflow:ConfigProto&GPU
2017-08-26 09:49
435 查看
tf.ConfigProto一般用在创建session的时候。用来对session进行参数配置
控制GPU资源使用率
控制使用哪块GPU
~/ CUDA_VISIBLE_DEVICES=0 python your.py#使用GPU0
~/ CUDA_VISIBLE_DEVICES=0,1 python your.py#使用GPU0,1
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 growth config = tf.ConfigProto() config.gpu_options.allow_growth = True session = tf.Session(config=config, ...) # 使用allow_growth option,刚一开始分配少量的GPU容量,然后按需慢慢的增加,由于不会释放 #内存,所以会导致碎片 # 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%
控制使用哪块GPU
~/ CUDA_VISIBLE_DEVICES=0 python your.py#使用GPU0
~/ CUDA_VISIBLE_DEVICES=0,1 python your.py#使用GPU0,1
相关文章推荐
- tensorflow 使用ConfigProto&GPU
- tensorflow:ConfigProto&GPU
- TensorFlow--ConfigProto&GPU
- 1. tensorflow学习之ConfigProto&GPU
- tensorflow学习笔记:ConfigProto&GPU(转自http://blog.csdn.net/u012436149/article/details/53837651)
- tensorflow中使用tf.ConfigProto()配置Session运行参数&&GPU设备指定
- ConfigProto&GPU
- tensorflow学习笔记(二十五):ConfigProto&GPU
- (转载)tensorflow学习笔记(二十五):ConfigProto&GPU
- Tensorflow ConfigProto & inter_/intra_op_parallelism_threads 整理
- 170614 Windows10+Anaconda2/3+tensorflow-1.2.0rc2(cpu & gpu) 安装
- 配置pycharm远程调试的环境(服务器为linux系统)&&服务器上安装tensorflow-gpu及配置cuda环境&&相关问题的解决办法
- cocos2d-x, protobuf, no config.h, #error "No suitable threading library available."
- 【TensorFlow】Windows+Anaconda3+Tensorflow(CPU&GPU)+Pycharm 安装
- Windows下Tensorflow-GPU的安装,Tensorboard的使用,以及过程中遇到的各种坑
- prototype & __proto__
- caffe和tensorflow对GPU的使用
- Ubuntu16.04安装tensorflow_gpu教程
- tensorflow GPU显存控制
- Tensorflow 多GPU下,指定使用某一块GPU