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