python - 即使我在每次迭代结束时删除所有变量，python for 循环中的内存泄漏

Question

以下两个循环将使用内存，直到用完为止，但我不知道为什么。我在每次迭代结束时删除所有创建的变量，但它仍然泄漏。

!pip3 install cupy-cuda101
import cupy as cp
import numpy as np
from sklearn.preprocessing import PolynomialFeatures

xtrain = cp.asnumpy(cp.random.uniform(-1,1,size = (150000,50)))

for i in range(0,1000):
   weights = cp.random.uniform(-1,1,size = (1275,1000))
   
   for chunk in range(0,xtrain.shape[0],5000):
      xchunk = xtrain[chunk:chunk+5000,:]
      poly=PolynomialFeatures(interaction_only = True, include_bias = False)
      xchunk = cp.array(poly.fit_transform(xchunk))
      ranks = cp.matmul(xchunk,weights)
      del ranks, xchunk, poly
   del weights

xtrain 也只是浮点数据，介于 -1 和 1 之间。

Answer 1

这些行在每次迭代结束时插入修复它:

    cp.get_default_memory_pool().free_all_blocks()
    cp.get_default_pinned_memory_pool().free_all_blocks()