解決Keras使用GPU資源耗盡的問(wèn)題
我們?cè)谑褂肎PU資源進(jìn)行訓(xùn)練的時(shí)候，可能會(huì)發(fā)生資源耗盡的情況，那么在在這種情況，我們需要對(duì)GPU的資源進(jìn)行合理的安排，具體使用辦法如下：
框架：Tensorflow和Keras
方法
import tensorflow as tf import keras.backend.tensorflow_backend as KTF config = tf.ConfigProto() config.gpu_options.allow_growth=True #不全部占滿顯存, 按需分配 sess = tf.Session(config=config) KTF.set_session(sess) # 設(shè)置session
通過(guò)這種方法，就能合理的使用GPU資源了。
至少到目前位置，我自己從程序沒(méi)出現(xiàn)資源耗盡的情況，當(dāng)然，對(duì)于batchsize的設(shè)置，一定要設(shè)置在合理的范圍，所謂合理，大家自己體會(huì)。
補(bǔ)充知識(shí)：keras使用GPU的一些坑
keras安裝
conda install tensorflow-gpu
pip install keras
注意：不要使用conda安裝keras，會(huì)檢測(cè)依賴，默認(rèn)安裝上tensorflow的CPU版本。
keras運(yùn)行報(bào)錯(cuò)，tensorflow/stream_executor/cuda/cuda_dnn.cc:329 Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
解決方法：
增加如下代碼：
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.5) config = tf.ConfigProto(gpu_options=gpu_options) config.gpu_options.allow_growth = True session = tf.Session(config=config)
以上這篇解決Keras使用GPU資源耗盡的問(wèn)題就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
本文給大家分享的是兩個(gè)分別使用python和php實(shí)現(xiàn)的將文件分割成小文件的代碼，非常的實(shí)用有需要的小伙伴可以參考下2017-03-03
這篇文章主要介紹了詳解numpy.meshgrid()方法使用，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2019-08-08
Python將數(shù)據(jù)生成二維碼的四種方法實(shí)例代碼
二維碼在日常生活中非常常見(jiàn),廣泛應(yīng)用于支付、登錄驗(yàn)證、信息分享等場(chǎng)景,下面這篇文章主要給大家介紹了關(guān)于Python將數(shù)據(jù)生成二維碼的四種方法,文中通過(guò)代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-09-09
這篇文章主要介紹了python3抓取中文網(wǎng)頁(yè)的方法,實(shí)例分析了Python3頁(yè)面抓取及編碼轉(zhuǎn)換的相關(guān)技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-07-07
使用Pytorch導(dǎo)出自定義ONNX算子的示例代碼
這篇文章主要介紹了使用Pytorch導(dǎo)出自定義ONNX算子的示例代碼,下面給出個(gè)具體應(yīng)用中的示例：需要導(dǎo)出pytorch的affine_grid算子,但在pytorch的2.0.1版本中又無(wú)法正常導(dǎo)出該算子,故可通過(guò)如下自定義算子代碼導(dǎo)出,需要的朋友可以參考下2024-03-03
pandas使用dtype/dtypes修改數(shù)據(jù)類型
在數(shù)據(jù)處理和分析中,經(jīng)常需要對(duì)數(shù)據(jù)進(jìn)行類型轉(zhuǎn)換以滿足特定的需求,本文主要介紹了pandas使用dtype/dtypes修改數(shù)據(jù)類型,具有一定的參考價(jià)值,感興趣的可以了解一下2024-06-06
這篇文章主要介紹了如何用python寫(xiě)個(gè)模板引擎，幫助大家更好的理解和使用python，感興趣的朋友可以了解下2021-01-01
python中sort()和sorted()的區(qū)別及用法實(shí)例
我們通常會(huì)遇到對(duì)數(shù)據(jù)庫(kù)中的數(shù)據(jù)進(jìn)行排序的問(wèn)題,下面這篇文章主要給大家介紹了關(guān)于python中sort()和sorted()的區(qū)別及用法的相關(guān)資料,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-06-06
Python利用ElementTree模塊處理XML的方法詳解
ElementTree是python的XML處理模塊，它提供了一個(gè)輕量級(jí)的對(duì)象模，下面這篇文章就來(lái)給大家介紹了關(guān)于Python利用ElementTree模塊處理XML的方法，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面來(lái)一起看看吧。2017-08-08
熱力圖，是一種通過(guò)對(duì)色塊著色來(lái)顯示數(shù)據(jù)的統(tǒng)計(jì)圖表。繪圖時(shí)，需指定顏色映射的規(guī)則。本文主要用Python來(lái)實(shí)現(xiàn)熱力圖的制作，需要的可以參考一下2022-09-09