tensorflow显存设置

为避免训练时占用全部的gpu显存，只需设置如下代码

import numpy as np
import tensorflow as tf
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '0'
# 程序最多占用制定gpu的50%显存
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.9)
# 程序按需申请内存
gpu_options.allow_growth = True
config = tf.ConfigProto(gpu_options=gpu_options)

init = tf.global_variables_initializer()
sess = tf.Session(config=config)

sess.close()