如何创建大 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)
或您需要的任何类型。这可能会节省内存。
当我尝试在 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)
或您需要的任何类型。这可能会节省内存。