#指定使用那块GPU训练
os.environ[“CUDA_VISIBLE_DEVICES”] = ‘0’
config = tf.ConfigProto()
设置最大占有GPU不超过显存的70%
config.gpu_options.per_process_gpu_memory_fraction = 0.7
重点:设置动态分配GPU
config.gpu_options.allow_growth = True
创建session时
with tf.Session(config=config) as sess: