报错解决：ResourceExhaustedError: OOM when allocating tensor with shape

最新推荐文章于 2024-07-07 17:57:43 发布

zhangpeterx

最新推荐文章于 2024-07-07 17:57:43 发布

阅读量2.4w

点赞数 5

分类专栏： python 机器学习行走的问题解决机

本文链接：https://blog.csdn.net/zhangpeterx/article/details/89292684

版权

在使用TensorFlow时遇到'ResourceExhaustedError: OOM when allocating tensor with shape'的错误，该错误是因为数据量过大无法一次性加载到GPU内存。解决方法是改用CPU进行训练，或者调整batch_size以优化GPU内存使用。

摘要由CSDN通过智能技术生成

报错解决：ResourceExhaustedError: OOM when allocating tensor with shape

早上在使用tensorflow时遇到如下报错：

Traceback (most recent call last):


  File "C:\Users\peter\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 786, in runfile
    execfile(filename, namespace)

  File "C:\Users\peter\Anaconda3\lib\site-packages\spyder_kernels\customize\spydercustomize.py", line 110, in execfile
    exec(compile