在AWS中使用Tensorflow导致资源耗尽错误

时间:2018-12-27 21:03:38

标签: python amazon-web-services tensorflow

我正在尝试在aws上运行基本的tensorflow lstm。我有一个使用tensorflow设置的基本EC2 ubuntu实例,以及要运行的python脚本。我的脚本一直进行训练(可以制作模型和处理数据),但是却给了我这个错误:tensorflow.python.framework.errors_impl.ResourceExhaustedError:OOM在分配带有shape [1024,1024]的张量并在/上键入float时作业:本地主机/副本:0 /任务:0 /设备:CPU:0通过分配器cpu

我认为这可能是EC2设置错误,因为相同的代码在我自己的机器上工作(尽管我使用的是jupyter笔记本)。但是,这是我第一次使用aws,因此我不熟悉它。 我的AWS设置是否有此错误,如果是,该如何解决?

0 个答案:

没有答案