【发布时间】：2013-06-09 17:28:33
【问题描述】：
下面的代码填满了我的所有记忆：
from sys import getsizeof
import numpy
# from http://stackoverflow.com/a/2117379/272471
def getSize(array):
return getsizeof(array) + len(array) * getsizeof(array[0])
class test():
def __init__(self):
pass
def t(self):
temp = numpy.zeros([200,100,100])
A = numpy.zeros([200], dtype = numpy.float64)
for i in range(200):
A[i] = numpy.sum( temp[i].diagonal() )
return A
a = test()
memory_usage("before")
c = [a.t() for i in range(100)]
del a
memory_usage("After")
print("Size of c:", float(getSize(c))/1000.0)
输出是：
('>', 'before', 'memory:', 20588, 'KiB ')
('>', 'After', 'memory:', 1583456, 'KiB ')
('Size of c:', 8.92)
如果 c 约为 9 KiB，为什么我要使用 ~1.5 GB 的内存？这是内存泄漏吗？ （谢谢）
memory_usage 函数已发布在 SO 上，为清楚起见，在此报告：
def memory_usage(text = ''):
"""Memory usage of the current process in kilobytes."""
status = None
result = {'peak': 0, 'rss': 0}
try:
# This will only work on systems with a /proc file system
# (like Linux).
status = open('/proc/self/status')
for line in status:
parts = line.split()
key = parts[0][2:-1].lower()
if key in result:
result[key] = int(parts[1])
finally:
if status is not None:
status.close()
print('>', text, 'memory:', result['rss'], 'KiB ')
return result['rss']
【问题讨论】：
-
用
float(getSize(c))/1000.0)替换float(getsizeof(c))/1000.0)，你应该会得到更好的价值。 -
感谢glglgl，我实际上复制了错误版本的示例代码。已更正。
-
如果在 t() 中的 return 语句之前放入 del temp 会发生什么？
-
不起作用。正如我所写，问题来自 numpy v. 1.7.0 的对角函数中的内存泄漏。已在 v 1.7.1 中修复。
标签： python memory-leaks numpy