Python numpy大矩陣運(yùn)算內(nèi)存不足如何解決
程序運(yùn)行,產(chǎn)生如下結(jié)果,然后進(jìn)程終止,導(dǎo)致這一結(jié)果的原因很有可能是內(nèi)存爆炸。
當(dāng)兩個較大的 (e.g., 10000*10000 維)ndarray 做運(yùn)算(加法,or 乘法)時,很容易出現(xiàn)這樣的結(jié)果.解決辦法:
大多數(shù)情況下,這種大矩陣都是稀疏的。盡可能地利用稀疏計(jì)算的方式,例如稀疏矩陣,或者只計(jì)算非 0 位置的值。 如果都是整數(shù)運(yùn)算,可以設(shè)置 dtype=int,而非 dtype=float, 可以省下不少空間。linux 系統(tǒng)下,使用 top 命令,可以很容易地看到內(nèi)存(%MEM) 的使用情況。
# 代碼段 1, true_similarity_matrix 是 int, similarity_matrix 是 floattmp_matrix = similarity_matrix * true_similarity_matrix # 內(nèi)存會炸掉,兩個 10000*10000 維 float arraynum_correct_edge = sum(sum(tmp_matrix))# 代碼段 2for i in range(): for j in range(): set_true_ij.append(i,j)num_correct_edge = 0for i, j in set_true_ij: num_correct_edge += similarity_matrix[i,j]
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持好吧啦網(wǎng)。
相關(guān)文章:
