关于使用tensorflow内存不够和gpu的问题

我安装了cuda及cudnn相关，有以下警告
2022-04-14 11:22:28.071802: I tensorflow/core/platform/cpu_feature_guard.cc:151] This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN) to use the following CPU instructions in performance-critical operations: AVX AVX2
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.
2022-04-14 11:22:28.944385: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1525] Created device /job:localhost/replica:0/task:0/device:GPU:0 with 3497 MB memory: -> device: 0, name: NVIDIA GeForce RTX 3060 Laptop GPU, pci bus id: 0000:01:00.0, compute capability: 8.6
Model: "note_classification_model"

然后开始训练以后，有以下错误，然后就自动退出了

img

我试过了减小batch_size，仍然没有作用，请问一下应该怎么解决，谢谢！

batch_size 降到最低试试 比如说1 2 4
subdivisions尽量大