tensorflow:OOM when allocating tensor with shape[225,256,256,36] and type float on /job:localhost/re

2021-10-26 111 阅读1分钟

运行TensorFlow（GPU）报错如下：

OOM when allocating tensor with shape[225,256,256,36] and type float on /job:localhost/replica:0/tas

原因是GPU OOM内存不够，因此可尝试改成批处理。即将训练的数据块调小。