Как создать большие матрицы np.zeros? - PullRequest
0 голосов
/ 12 мая 2019

Когда я пытаюсь создать пустую матрицу в Colab с 12 ГБ ОЗУ, я вижу ошибку памяти:

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

Как я могу это исправить?

1 Ответ

1 голос
/ 12 мая 2019

Тип по умолчанию: np.float64, то есть 8 байт. Размер в ГБ составляет 644253 * 50 * 259 * 8/1024 ** 3 = 62,16 ГБ. Вы не можете создать такую ​​матрицу, недостаточно памяти. Если вам это действительно нужно, добавьте много памяти подкачки, но она будет очень медленной по сравнению с ОЗУ (может быть на несколько порядков медленнее). Вы также можете попробовать A = np.zeros((644253, 50, 259), type=np.float32) или A = np.zeros((644253, 50, 259), type=np.int8) или любой другой тип, который вам нужен. Это может сэкономить память.

...