GPU ResourceExhaustedError

Question

我正在从头训练VGG16网络以解决二进制图像分类问题，但不幸的是，我遇到此错误：

ResourceExhaustedError:  OOM when allocating tensor with shape[401408,1024] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
     [[node gradients/dense_1/MatMul_grad/MatMul_1 (defined at /opt/conda/lib/python3.7/site-packages/keras/backend/tensorflow_backend.py:3009) ]]
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
 [Op:__inference_keras_scratch_graph_4981]

Function call stack:
keras_scratch_graph

根据我对该错误的一些建议，将批量大小从64减少到16，但是没有运气。

我的VM具有8GB ROM和4个GPU，数据大小小于10MB。而GPU的细节如下：