Tensorflow 實現(xiàn)釋放內(nèi)存
更新時間：2020年02月03日 10:17:03 作者：榮軒浩
今天小編就為大家分享一篇Tensorflow 實現(xiàn)釋放內(nèi)存，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
我在做tensorflow開發(fā)的時候遇到如下的錯誤，我請教了深度學習社區(qū)q群中的大神，大神說這是運算圖還在內(nèi)存中，沒有及時釋放，需要restart一下kernel，但是由于我的代碼不能停止執(zhí)行，所以沒辦法用重新運行代碼來解決釋放內(nèi)存運行圖的問題。
問題：
with tf.Session() as sess: saver = tf.train.Saver() #保存運算圖
當我退出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()
代碼來重置默認的圖，這樣就能解決我下一步執(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)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關文章
終端能到import模塊 解決jupyter notebook無法導入的問題
這篇文章主要介紹了在終端能到import模塊 而在jupyter notebook無法導入的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
基于Python和Tkinter實現(xiàn)高考倒計時功能
隨著高考的臨近,每個考生都在緊鑼密鼓地復習,這時候,一款實用的倒計時軟件能有效幫助你規(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檢索文件的操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2021-05-05
在Python中，判斷一個列表（List）是否包含某個特定元素是常見的任務之一，本文就來介紹一下多種判斷List成員包含性的方法，并提供豐富的示例代碼，以幫助大家更好地理解和運用這些技術2023-12-12
python腳本實現(xiàn)分析dns日志并對受訪域名排行
這篇文章主要介紹了python腳本實現(xiàn)分析dns日志并對受訪域名排行,本文是在Windows服務器環(huán)境中實現(xiàn),需要的朋友可以參考下2014-09-09
多個geojson經(jīng)過坐標系轉(zhuǎn)換后如何合并為一個shp數(shù)據(jù)
這篇文章主要介紹了多個geojson經(jīng)過坐標系轉(zhuǎn)換后如何合并為一個shp數(shù)據(jù)問題,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-10-10
這篇文章主要為大家介紹了Python全棧之MySQL，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-01-01
這篇文章主要為大家介紹了python密碼學周期置換密碼的學習，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-05-05