GPU memory leak

[wangsssky]

It runs into an error when I try to predict images in a loop.
File "cupy/core/_kernel.pyx", line 801, in cupy.core._kernel.ufunc.__call__ File "cupy/core/_kernel.pyx", line 329, in cupy.core._kernel._get_out_args File "cupy/core/core.pyx", line 150, in cupy.core.core.ndarray.__init__ File "cupy/cuda/memory.pyx", line 517, in cupy.cuda.memory.alloc File "cupy/cuda/memory.pyx", line 1064, in cupy.cuda.memory.MemoryPool.malloc File "cupy/cuda/memory.pyx", line 1085, in cupy.cuda.memory.MemoryPool.malloc File "cupy/cuda/memory.pyx", line 899, in cupy.cuda.memory.SingleDeviceMemoryPool.malloc File "cupy/cuda/memory.pyx", line 920, in cupy.cuda.memory.SingleDeviceMemoryPool._malloc File "cupy/cuda/memory.pyx", line 694, in cupy.cuda.memory._try_malloc cupy.cuda.memory.OutOfMemoryError: out of memory to allocate 154880000 bytes (total 7112607232 bytes)