解決Tensorflow占用GPU顯存問(wèn)題
我使用Pytorch進(jìn)行模型訓(xùn)練時(shí)發(fā)現(xiàn)真正模型本身對(duì)于顯存的占用并不明顯，但是對(duì)應(yīng)的轉(zhuǎn)換為tensorflow后（權(quán)重也進(jìn)行了轉(zhuǎn)換），發(fā)現(xiàn)Python-tensorflow在使用時(shí)默認(rèn)吃掉所有顯存，并且不手動(dòng)終結(jié)程序的話顯存并不釋放（我有兩個(gè)序貫的模型，前面一個(gè)跑完后并不釋放占用顯存）（https://github.com/tensorflow/tensorflow/issues/1727），這一點(diǎn)對(duì)于后續(xù)的工作有很大的影響。
后面發(fā)現(xiàn)python-tensorflow限制顯存有兩種方法：
1. 設(shè)置顯卡的使用率
這種方法在學(xué)習(xí)和工作中比較好用，學(xué)習(xí)時(shí)可提高顯卡使用效率，工作時(shí)可方便的獲得GPU顯存消耗極限，用以提供顯卡購(gòu)買時(shí)的參數(shù)，現(xiàn)將代碼展示如下：
這里的0.1 表示使用顯存總量的的10%
2. 設(shè)置顯卡按需使用（這個(gè)本人并沒(méi)有專門測(cè)試，只是從tensorflow論壇上獲得）
gpu_options = tf.GPUOptions(allow_growth=True) sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
以上這篇解決Tensorflow占用GPU顯存問(wèn)題就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
pandas.DataFrame刪除/選取含有特定數(shù)值的行或列實(shí)例
今天小編就為大家分享一篇pandas.DataFrame刪除/選取含有特定數(shù)值的行或列實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-11-11
Python使用difflib標(biāo)準(zhǔn)庫(kù)實(shí)現(xiàn)查找文本間的差異
在文本處理和比較中,查找文本之間的差異是一項(xiàng)常見(jiàn)的任務(wù),本文將詳細(xì)介紹如何使用difflib模塊來(lái)查找文本之間的差異,包括單行和多行文本的比較、生成差異報(bào)告,需要的可以參考下2024-03-03
這篇文章主要給大家介紹了關(guān)于Python中zipfile壓縮文件模塊的基本使用教程，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家學(xué)習(xí)或者使用Python具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-06-06
基于Python實(shí)現(xiàn)圖片瀏覽器的應(yīng)用程序
圖像瀏覽器應(yīng)用程序是一種非常常見(jiàn)和實(shí)用的工具,這篇文章就來(lái)為大家介紹一下如何使用Python編程語(yǔ)言和wxPython庫(kù)創(chuàng)建一個(gè)簡(jiǎn)單的圖像瀏覽器應(yīng)用程序,感興趣的可以了解下2023-10-10
Python 調(diào)用DLL的實(shí)用代碼。2009-01-01
這篇文章主要介紹了利用Python實(shí)現(xiàn)kNN算法的代碼，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08
這篇文章主要為大家介紹了Python?日志記錄模塊的綜合指南,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-12-12
這篇文章主要給大家介紹了關(guān)于django中cookiecutter使用的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2020-12-12
Python數(shù)據(jù)結(jié)構(gòu)之Array用法實(shí)例
這篇文章主要介紹了Python數(shù)據(jù)結(jié)構(gòu)之Array用法實(shí)例,較為詳細(xì)的講述了Array的常見(jiàn)用法,具有很好的參考借鑒價(jià)值,需要的朋友可以參考下2014-10-10
python選取特定列 pandas iloc,loc,icol的使用詳解(列切片及行切片)
今天小編就為大家分享一篇python選取特定列 pandas iloc,loc,icol的使用詳解(列切片及行切片)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2019-08-08