0
点赞
收藏
分享

微信扫一扫

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


运行TensorFlow(GPU)报错如下:

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

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

参考链接:
​​​https://stackoverflow.com/questions/39076388/tensorflow-deep-mnist-resource-exhausted-oom-when-allocating-tensor-with-shape​​

​​https://github.com/tensorflow/tensorflow/issues/609​​


举报

相关推荐

0 条评论