如何创建大 np.zeros 矩阵?

How to create big np.zeros matrices?

当我尝试在 Colab 中使用 12GB RAM 创建一个 numpy 矩阵时,我看到内存错误:

A = np.zeros((644253, 50, 259))

我该如何解决这个问题?

默认类型为np.float64,8字节。以 GB 为单位的大小为 644253*50*259*8/1024**3 = 62.16GB。你不能创建这样的矩阵,没有足够的内存。如果你真的需要它,添加大量交换内存,但与 RAM 相比它会非常慢(可能慢几个数量级)。您也可以尝试 A = np.zeros((644253, 50, 259), type=np.float32)A = np.zeros((644253, 50, 259), type=np.int8) 或您需要的任何类型。这可能会节省内存。