tensorflow 限制顯存大小的實現(xiàn)
更新時間：2020年02月03日 11:52:31 作者：安陽小棧-客官歇會吧
今天小編就為大家分享一篇tensorflow 限制顯存大小的實現(xiàn)，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
Python在用GPU跑模型的時候最好開多進(jìn)程，因為很明顯這種任務(wù)就是計算密集型的。
用進(jìn)程池好管理，但是tensorflow默認(rèn)情況會最大占用顯存，盡管該任務(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)文章
numpy 進(jìn)行數(shù)組拼接,分別在行和列上合并的實例
今天小編就為大家分享一篇numpy 進(jìn)行數(shù)組拼接,分別在行和列上合并的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
Pytest+request+Allure實現(xiàn)接口自動化框架
接口自動化是指模擬程序接口層面的自動化，由于接口不易變更，維護(hù)成本更小，所以深受各大公司的喜愛，本文主要介紹了Pytest+request+Allure實現(xiàn)接口自動化框架，感興趣的可以了解一下2021-07-07
Python語言實現(xiàn)獲取主機名根據(jù)端口殺死進(jìn)程
這篇文章主要介紹了Python語言實現(xiàn)獲取主機名根據(jù)端口殺死進(jìn)程的相關(guān)資料,需要的朋友可以參考下2016-03-03
大家好，本篇文章主要講的是用python實現(xiàn)谷歌小恐龍小游戲，看看這是你斷網(wǎng)時的樣子么，感興趣的同學(xué)趕快來看一看吧，對你有幫助的話記得收藏一下2022-02-02
這篇文章主要介紹了Python 并行化執(zhí)行詳細(xì)解析，文章圍繞主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價值，需要的小伙伴可以參考一下，希望對你的學(xué)習(xí)有所幫助2022-07-07
Python編程實現(xiàn)二分法和牛頓迭代法求平方根代碼
這篇文章主要介紹了Python編程實現(xiàn)二分法和牛頓迭代法求平方根代碼，具有一定參考價值，需要的朋友可以了解下。2017-12-12
今天小編就為大家分享一篇對python修改xml文件的節(jié)點值方法詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-12-12
pycharm自定義TODO類注釋以及高亮顏色的設(shè)置方法
這篇文章主要介紹了pycharm自定義TODO類注釋以及高亮顏色的設(shè)置方法,文中通過圖文結(jié)合的方式給大家介紹的非常詳細(xì),具有一定的參考價值,需要的朋友可以參考下2024-03-03
python實現(xiàn)帶驗證碼網(wǎng)站的自動登陸實現(xiàn)代碼
本例所登錄的某網(wǎng)站需要提供用戶名，密碼和驗證碼，在此使用了python的urllib2直接登錄網(wǎng)站并處理網(wǎng)站的Cookie2015-01-01
詳解TensorFlow訓(xùn)練網(wǎng)絡(luò)兩種方式
本文主要介紹了TensorFlow訓(xùn)練網(wǎng)絡(luò)兩種方式，一種是基于tensor(array)，另外一種是迭代器，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-12-12