Tensorflow 實(shí)現(xiàn)釋放內(nèi)存
我在做tensorflow開發(fā)的時候遇到如下的錯誤，我請教了深度學(xué)習(xí)社區(qū)q群中的大神，大神說這是運(yùn)算圖還在內(nèi)存中，沒有及時釋放，需要restart一下kernel，但是由于我的代碼不能停止執(zhí)行，所以沒辦法用重新運(yùn)行代碼來解決釋放內(nèi)存運(yùn)行圖的問題。
問題：
with tf.Session() as sess: saver = tf.train.Saver() #保存運(yùn)算圖
當(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)存的運(yùn)算圖就產(chǎn)生了報(bào)錯，
解決辦法：
在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 實(shí)現(xiàn)釋放內(nèi)存就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
終端能到import模塊 解決jupyter notebook無法導(dǎo)入的問題
這篇文章主要介紹了在終端能到import模塊 而在jupyter notebook無法導(dǎo)入的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
基于Python和Tkinter實(shí)現(xiàn)高考倒計(jì)時功能
隨著高考的臨近,每個考生都在緊鑼密鼓地復(fù)習(xí),這時候,一款實(shí)用的倒計(jì)時軟件能有效幫助你規(guī)劃剩余時間,提醒你不要浪費(fèi)每一分每一秒,今天,我們來聊聊一款基于Python和Tkinter開發(fā)的高考倒計(jì)時軟件,功能簡單卻極具實(shí)用性,讓你在緊張的備考過程中不再迷失2025-03-03
macOS M1(Apple Silicon)安裝配置Conda環(huán)境的具體實(shí)現(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）是否包含某個特定元素是常見的任務(wù)之一，本文就來介紹一下多種判斷List成員包含性的方法，并提供豐富的示例代碼，以幫助大家更好地理解和運(yùn)用這些技術(shù)2023-12-12
python腳本實(shí)現(xiàn)分析dns日志并對受訪域名排行
這篇文章主要介紹了python腳本實(shí)現(xiàn)分析dns日志并對受訪域名排行,本文是在Windows服務(wù)器環(huán)境中實(shí)現(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全棧之MySQL，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2022-01-01
python密碼學(xué)周期置換密碼學(xué)習(xí)
這篇文章主要為大家介紹了python密碼學(xué)周期置換密碼的學(xué)習(xí)，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-05-05