深度学习用tensorflow,在进行训练一段时间后,会出现“OOM when allocating tensor with shape[9,256,94,167] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc”
显卡GPU显存不够用了,建议把batch-size 调小一点
深度学习用tensorflow,在进行训练一段时间后,会出现“OOM when allocating tensor with shape[9,256,94,167] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc”
显卡GPU显存不够用了,建议把batch-size 调小一点