遇到这个问题,是要初始化一个大的矩阵:
import numpy as np
init_a = np.zeros((10000*10000,4096))
直接这样初始化就提示MemoryError.
查资料发现默认的dtype=float64
于是修改数据类型为float16
之后,避免了错误,虽然对损失了精度,但代码运行成功了:
init_a = np.zeros((10000*10000,4096),dtype='float16')
如果有更好的解决办法,欢迎指教~