MemoryError tensorflow

时间:2017-04-25 16:36:59

标签: python python-2.7 amazon-web-services tensorflow p2

我正在从P2.xlarge类型的AWS实例运行此模型。它给出了一个错误:

Exception in thread Thread-16:
Traceback (most recent call last):
File "/usr/lib/python2.7/threading.py", line 801, in __bootstrap_inner
self.run()
File "/usr/lib/python2.7/threading.py", line 754, in run
self.__target(*self.__args, **self.__kwargs)
File "/home/ubuntu/tensorflow/models/summarization/textsum/batch_reader.py" , line 136, in _FillInputQueue
(article, abstract) = input_gen.next()
File "/home/ubuntu/tensorflow/models/summarization/textsum/batch_reader.py", line 245, in _TextGenerator
e = example_gen.next()
File "/home/ubuntu/tensorflow/models/summarization/textsum/data.py", line 109, in ExampleGen
example_str = struct.unpack('%ds' % str_len, reader.read(str_len))[0]
MemoryError

系统存储信息是 -

Filesystem Size Used Avail Use% Mounted on
udev 30G 0 30G 0% /dev
tmpfs 6.0G 8.9M 6.0G 1% /run
/dev/xvda1 30G 12G 18G 39% /
tmpfs 30G 0 30G 0% /dev/shm
tmpfs 5.0M 0 5.0M 0% /run/lock
tmpfs 30G 0 30G 0% /sys/fs/cgroup
tmpfs 6.0G 0 6.0G 0% /run/user/1000

NVIDIA状态 -

ubuntu@ip-172-31-28-161:~$ lspci | grep -i nvidia

00:1e.0 3D控制器:NVIDIA Corporation GK210GL [特斯拉K80](rev a1)

这是什么解决方案?

如果我将str_len = struct.unpack('q', len_bytes)[0]替换为str_len = struct.unpack('Bi', len_bytes)[0] 然后这个错误就消失了,新错误出现了:

Exception in thread Thread-15:
Traceback (most recent call last):
File "/usr/lib/python2.7/threading.py", line 801, in __bootstrap_inner
self.run()
File "/usr/lib/python2.7/threading.py", line 754, in run
self.__target(*self.__args, **self.__kwargs)
File "/home/mindstix/bazel/models/Summarizer/textsum/batch_reader.py", line 136, in _FillInputQueue
(article, abstract) = input_gen.next()
File "/home/mindstix/bazel/models/Summarizer/textsum/batch_reader.py", line 248, in _TextGenerator
article_text = self._GetExFeatureText(e, self._article_key)
File "/home/mindstix/bazel/models/Summarizer/textsum/batch_reader.py", line 265, in _GetExFeatureText
return ex.features.feature[key].bytes_list.value[0]
IndexError: list index (0) out of range

如果我打印example_str,则屏幕上会显示该值。但是当我尝试打印ex.features.feature[key].bytes_list.value时,它会返回空白。

应该采取什么措施来解决这一切?

这是我遵循的代码步骤:

>>> import tensorflow as tf
>>> import struct
>>>from tensorflow.core.example import example_pb2
>>> reader = open('data/training-1', 'rb')
>>> len_bytes = reader.read(8)
>>> str_len = struct.unpack('q', len_bytes)[0]
>>> str_len
2335523720558635124
>>> example_str = struct.unpack('%ds' % str_len, reader.read(str_len))[0]
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
MemoryError

>>> str_len = struct.unpack('Bi', len_bytes)[0]
>>> str_len
116

>>> example_str = struct.unpack('%ds' % str_len, reader.read(str_len))[0]
>>>e = example_pb2.Example.FromString(example_str)
>>> e.features.feature['article'].bytes_list.value
<google.protobuf.pyext._message.RepeatedScalarContainer object at  0x7fc25c9325a8>

>>> e.features.feature['article'].bytes_list.value[0] 
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
IndexError: list index (0) out of range

1 个答案:

答案 0 :(得分:1)

我遇到了同样的问题。但原因是我使用原始文本文件进行测试。它应该是使用传输的二进制文件。我不确定你的情况是否和我一样。