keras Tensoflow CPU:分配超过系统内存的10%

amrnrhlw  于 2023-03-30  发布在  其他
关注(0)|答案(1)|浏览(140)

我使用Tensorflow(CNN+RNN模型)在Keras中实现了一个模型。然而,当我想评估我的模型时,我有以下错误:

2018󈚰󈚪 09:48:44.961161: I tensorflow/core/platform     
/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA

2018󈚰󈚪 09:49:00.450298: W tensorflow/core/framework/allocator.cc:101]    
Allocation of 3359939800 exceeds 10% of system memory.

terminate called after throwing an instance of 'std::bad_alloc'
what(): std::bad_alloc

这是一个CPU内存分配的问题,但我不知道如何解决这个问题。我的模型参数很小,所以我不认为这是问题。谢谢你的帮助。

0kjbasz6

0kjbasz61#

您可以尝试减小输入数据的批处理大小或图像大小,或者使用较小的模型。此外,您还可以尝试通过添加更多RAM或GPU或更多内存机器来增加系统上的可用内存量

相关问题