config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
当allow_growth设置为True时,分配器将不会指定所有的GPU内存,而是根据需求增长(与theano类似)。
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config, ...)
当allow_growth设置为True时,分配器将不会指定所有的GPU内存,而是根据需求增长(与theano类似)。