我已经使用Tensorflow Slim制作了DiscoGAN。代码在CPU上执行,但在尝试在GPU上运行时会引发内存不足错误。堆栈跟踪在一次执行中多次显示以下错误:
Allocator (GPU_0_bfc) ran out of memory trying to allocate 288.00MiB. Current allocation summary follows.
该错误的大小会有所不同(请参阅下面的完整堆栈跟踪) - 有时它甚至不会达到288 MB。我看到4个字节显示为错误。
我现在正在测试执行。我的批量大小为12(超过72张图像),每张图像为64x64x3。
其他一些有类似错误的人设法通过在代码中添加以下内容来解决此问题:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config = config)
然而,这并没有解决我的问题。
堆栈跟踪是:https://pastebin.com/qeisC6PC(太大而无法在此处粘贴)