tensorflow默认占用全部gpu内存,这样会出错
设置每个GPU内存占用率70%
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7)
config=tf.ConfigProto(gpu_options=gpu_options)
sess = tf.Session(config=config)
ps:具体在一个程序中,只需将前两句放到import一堆后面的位置,当全局变量
然后查找所有的tf.Session,加入config=config的参数