TensorFlow中出现 CUDA_ERROR_OUT_OF_MEMORY

参考
https://blog.csdn.net/qq_34914551/article/details/86604988

在定义session之前，增加
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config = tf.ConfigProto(log_device_placement=False,gpu_options=gpu_options)
with tf.Session(config=config)
 
 
也可以
config.gpu_options.allow_growth=True

我设置后还是没有解决问题，又把batch_size改小了点才解决