tensorflow 限制顯存大小的實現(xiàn)
更新時間：2020年02月03日 11:52:31 作者：安陽小棧-客官歇會吧
今天小編就為大家分享一篇tensorflow 限制顯存大小的實現(xiàn)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
Python在用GPU跑模型的時候最好開多進程，因為很明顯這種任務(wù)就是計算密集型的。
用進程池好管理，但是tensorflow默認情況會最大占用顯存，盡管該任務(wù)并不需要這么多，因此我們可以設(shè)置顯存的按需獲取，這樣程序就不會死掉了。
1. 按比例預(yù)留：
tf_config = tensorflow.ConfigProto() tf_config.gpu_options.per_process_gpu_memory_fraction = 0.5 # 分配50% session = tensorflow.Session(config=tf_config)
2. 或者干脆自適應(yīng)然后自動增長：
tf_config = tensorflow.ConfigProto() tf_config.gpu_options.allow_growth = True # 自適應(yīng) session = tensorflow.Session(config=tf_config)
以上這篇tensorflow 限制顯存大小的實現(xiàn)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python bsddb模塊操作Berkeley DB數(shù)據(jù)庫介紹
這篇文章主要介紹了Python bsddb模塊操作Berkeley DB數(shù)據(jù)庫介紹,這里簡單介紹一些關(guān)于bsddb的使用方法,需要的朋友可以參考下2015-04-04
這篇文章主要介紹了pycharm顯示行號的方法,需要的朋友可以參考下2014-06-06
python??UPX?is?not?available問題解決方法
這篇文章主要介紹了python?UPX?is?not?available問題解決,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2023-04-04
Python3基礎(chǔ)教程之遞歸函數(shù)簡單示例
這篇文章主要給大家介紹了關(guān)于Python3基礎(chǔ)教程之遞歸函數(shù)簡單示例的相關(guān)資料，文中通過示例代碼介紹的非常詳細，對大家學習或者使用Python3具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧2019-06-06
這篇文章主要為大家詳細介紹了opencv實現(xiàn)圖片模糊和銳化操作，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-11-11
這篇文章主要介紹了python制作一個解壓縮軟件的方法，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下2019-11-11
python登錄QQ郵箱發(fā)信的代碼，有需要的朋友可以參考下2013-02-02
這篇文章主要介紹了Python?如何實現(xiàn)變量交換，Python?程序員肯定知道?a,b?=?b,a，這句話用來交換兩個變量。相較于其它語言需要引入一個?temp?來臨時存儲變量的做法，Python?的這種寫法無疑非常優(yōu)雅，下面我們來看看具體的實現(xiàn)過程吧2022-01-01
這篇文章主要介紹了詳解python中各種的函數(shù),是Python入門中的基礎(chǔ)知識,需要的朋友可以參考下，希望能夠給你帶來幫助2021-10-10
python字符串分割常用方法(str.split()和正則)
在Python中字符串是一種非常常見的數(shù)據(jù)類型,在實際應(yīng)用中我們經(jīng)常需要對字符串進行分割,以便對其中的內(nèi)容進行處理,這篇文章主要給大家介紹了關(guān)于python字符串分割(str.split()和正則)的相關(guān)資料,需要的朋友可以參考下2023-11-11