Tensorflow 實現(xiàn)釋放內(nèi)存

更新時間：2020年02月03日 10:17:03 作者：榮軒浩

今天小編就為大家分享一篇Tensorflow 實現(xiàn)釋放內(nèi)存，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

我在做tensorflow開發(fā)的時候遇到如下的錯誤，我請教了深度學(xué)習(xí)社區(qū)q群中的大神，大神說這是運算圖還在內(nèi)存中，沒有及時釋放，需要restart一下kernel，但是由于我的代碼不能停止執(zhí)行，所以沒辦法用重新運行代碼來解決釋放內(nèi)存運行圖的問題。

問題：

with tf.Session() as sess:

saver = tf.train.Saver() #保存運算圖

當(dāng)我退出with并且立馬執(zhí)行

self.sess = tf.Session()
self.saver = tf.train.import_meta_graph('./Model/model.ckpt.meta')
self.saver.restore(self.sess, tf.train.latest_checkpoint('./Model/'))

的時候由于session和graph沒有釋放內(nèi)存的運算圖就產(chǎn)生了報錯，

解決辦法：

在with tf.Session() as sess: 之后同時也要在with的范圍以外（注意），添加

tf.reset_default_graph()

代碼來重置默認(rèn)的圖，這樣就能解決我下一步執(zhí)行代碼

self.sess = tf.Session()
self.saver = tf.train.import_meta_graph('./Model/model.ckpt.meta')
self.saver.restore(self.sess, tf.train.latest_checkpoint('./Model/'))

所產(chǎn)生的問題了。

以上這篇Tensorflow 實現(xiàn)釋放內(nèi)存就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

終端能到import模塊解決jupyter notebook無法導(dǎo)入的問題
這篇文章主要介紹了在終端能到import模塊而在jupyter notebook無法導(dǎo)入的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2021-03-03
基于Python和Tkinter實現(xiàn)高考倒計時功能
隨著高考的臨近,每個考生都在緊鑼密鼓地復(fù)習(xí),這時候,一款實用的倒計時軟件能有效幫助你規(guī)劃剩余時間,提醒你不要浪費每一分每一秒,今天,我們來聊聊一款基于Python和Tkinter開發(fā)的高考倒計時軟件,功能簡單卻極具實用性,讓你在緊張的備考過程中不再迷失
2025-03-03
macOS M1(Apple Silicon)安裝配置Conda環(huán)境的具體實現(xiàn)
由于常用的Anaconda和Miniconda現(xiàn)在都沒有提供M1處理器支持的conda環(huán)境，以下是conda-forge提供的miniforge，感興趣的可以了解一下
2021-08-08
tensorflow: variable的值與variable.read_value()的值區(qū)別詳解
今天小編就為大家分享一篇tensorflow: variable的值與variable.read_value()的值區(qū)別詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-07-07
python使用glob檢索文件的操作
這篇文章主要介紹了python使用glob檢索文件的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2021-05-05
Python判斷List中是否包含某個元素
在Python中，判斷一個列表（List）是否包含某個特定元素是常見的任務(wù)之一，本文就來介紹一下多種判斷List成員包含性的方法，并提供豐富的示例代碼，以幫助大家更好地理解和運用這些技術(shù)
2023-12-12
python腳本實現(xiàn)分析dns日志并對受訪域名排行
這篇文章主要介紹了python腳本實現(xiàn)分析dns日志并對受訪域名排行,本文是在Windows服務(wù)器環(huán)境中實現(xiàn),需要的朋友可以參考下
2014-09-09
多個geojson經(jīng)過坐標(biāo)系轉(zhuǎn)換后如何合并為一個shp數(shù)據(jù)
這篇文章主要介紹了多個geojson經(jīng)過坐標(biāo)系轉(zhuǎn)換后如何合并為一個shp數(shù)據(jù)問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教
2023-10-10
Python全棧之學(xué)習(xí)MySQL(1)
這篇文章主要為大家介紹了Python全棧之MySQL，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助
2022-01-01
python密碼學(xué)周期置換密碼學(xué)習(xí)
這篇文章主要為大家介紹了python密碼學(xué)周期置換密碼的學(xué)習(xí)，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2022-05-05