使用 GPU 的 Google Colab 上的 Cupy 内存错误 - 但这只是我第二次运行代码

问：

我正在尝试使用 Cupy 对两个大型数组进行矩阵乘法运算，因为它比使用 CPU 快得多（约 100 倍）。我的问题是它第一次运行它时可以工作，但第二次以此类推，它给了我一个内存错误。这是一个循环中的步骤，所以这是一个问题，我不能每次都重新启动运行时。

下面是具有相同数组大小和数据类型的可重现代码：

import cupy as cp
import datetime

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

x = cp.random.uniform(-1,1,size = (3000,300000))
w = cp.random.uniform(-1,1,size= (300000,1000))

start = datetime.now()
ans = cp.matmul(x,w)
stop = datetime.now()
print(stop-start)

这是我在同一运行时第二次运行它时收到的错误：

---------------------------------------------------------------------------
OutOfMemoryError                          Traceback (most recent call last)
<ipython-input-5-43db33b58bc8> in <module>()
      2 cp.get_default_pinned_memory_pool().free_all_blocks()
      3 
----> 4 x = cp.random.uniform(-1,1,size = (3000,300000))
      5 w = cp.random.uniform(-1,1,size= (300000,1000))
      6 

4 frames
/usr/local/lib/python3.6/dist-packages/cupy/creation/basic.py in empty(shape, dtype, order)
     20 
     21     """
---> 22     return cupy.ndarray(shape, dtype, order=order)
     23 
     24 

cupy/core/core.pyx in cupy.core.core.ndarray.__init__()

cupy/cuda/memory.pyx in cupy.cuda.memory.alloc()

cupy/cuda/memory.pyx in cupy.cuda.memory.MemoryPool.malloc()

cupy/cuda/memory.pyx in cupy.cuda.memory.MemoryPool.malloc()

cupy/cuda/memory.pyx in cupy.cuda.memory.SingleDeviceMemoryPool.malloc()

cupy/cuda/memory.pyx in cupy.cuda.memory.SingleDeviceMemoryPool._malloc()

cupy/cuda/memory.pyx in cupy.cuda.memory._try_malloc()

OutOfMemoryError: Out of memory allocating 7,200,000,000 bytes (allocated so far: 9,624,000,000 bytes).

这个问题可以解决吗？我正在尝试清除前两行中的 GPU 内存，但不确定这是否正确。也许使用 dask 数组会起作用？但是，在仍然使用 GPU 提高速度的同时，可以做到这一点吗？