Tensorflow 實現釋放記憶體

2022-09-27 08:39:10 字數 849 閱讀 4070

我在做tensorflow開發的時候遇到如下的錯誤,我請教了深度學習社群q群中的大神,大神說這是運算圖還在記憶體中,沒有及時釋放,需要restart一下kernel,但是由於我的**不能停止執行,所以沒辦法用重新執行**來解決釋放記憶體執行圖的問題。

問題:with tf.session() as sess:

s**er = tf.train.s**er() #儲存運算圖

當我退程式設計客棧出with並且立馬執行

self.sess = tf.session()

self.s**er = tf.train.import_meta_graph('./model/model.ckpt.meta')

self.s**er.restore(self.sess, tf.train.latest_checkpoin程式設計客棧t('./model/'))

的時候由於session和graph沒有釋放記憶體的運算圖就產生了報錯,

解決辦法:

在with tf.session() as sess 之後同時也要在with的範圍以外(注意),新增

tf.reset_default_graph()

**來重置預設的程式設計客棧圖,這樣就能解決我下一步執行**

s程式設計客棧elf.sess = tf.session()

self.s**er = tf.train.import_meta_graph('./model/model.ckpt.meta')

self.s**er.restore(self.sess, tf.train.latest_checkpoint('./model/'))

所產生的問題了。

本文標題: tensorflow 實現釋放記憶體

本文位址:

Objective C ARC自動釋放物件記憶體

arc是cocoa系統幫你完成物件記憶體釋放的引用計數機制 h檔案 1 01 arc3 4 created by ma c on 15 8 13.5 6 78 import910 inte ce person nsobject 11 property nonatomic,strong nsstrin...

python opencv下讀取影像釋放記憶體

因為讀取影像較大,且數量較多,不釋放,報錯opencv 4.1.1 4 insufficient memory failed to allocate 2825196594 bytes c 下opencv釋放影像占用記憶體 iplimage img cvcreateimage cvsize 640,4...

TensorFlow訓練內(顯)存不斷增長

在使用tensorflow過程中,乙個不標準的操作,就可能導致程式出各種bug,今天我們的豬腳就是 tensorflow訓練內 顯 存不斷增長 此問題並不是我遇到的,是公司一位同事遇到的,我把 翻了一下,看出了問題所在,由於一些保密原因,我就不在這裡展示那個 但可以用其他 來替代。import te...