运行AlexNet：cuda_error_out_of_memory错误

Question

Gpu内存：11.9G

Init alexnet：需要34G，因此内存不足。

使用tensorflow官方网络的两种方式之一如下，但仍然是旧问题。

config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.7
session = tf.Session(config=config) 

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
session = tf.Session(config=config)

为什么？任何好主意，非常感谢!!!

Answer 1

真正的原因是，在shuffle中min_after_dequeue太大了