解決Keras使用GPU資源耗盡的問題

更新時(shí)間：2020年06月22日 10:57:20 作者：wp1988

這篇文章主要介紹了解決Keras使用GPU資源耗盡的問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

我們?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

通過這種方法，就能合理的使用GPU資源了。

至少到目前位置，我自己從程序沒出現(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資源耗盡的問題就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Android申請(qǐng)相機(jī)權(quán)限和讀寫權(quán)限實(shí)例
大家好，本篇文章主要講的是Android申請(qǐng)相機(jī)權(quán)限和讀寫權(quán)限實(shí)例，感興趣的同學(xué)趕快來看一看吧，對(duì)你有幫助的話記得收藏一下
2022-02-02
python判斷字符串或者集合是否為空的實(shí)例
今天小編就為大家分享一篇python判斷字符串或者集合是否為空的實(shí)例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2019-01-01
使用Keras加載含有自定義層或函數(shù)的模型操作
這篇文章主要介紹了使用Keras加載含有自定義層或函數(shù)的模型操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2020-06-06
用python擬合等角螺線的實(shí)現(xiàn)示例
這篇文章主要介紹了用python擬合等角螺線的實(shí)現(xiàn)示例，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2019-12-12
Python中遍歷字典過程中更改元素導(dǎo)致異常的解決方法
這篇文章主要介紹了Python中遍歷字典過程中更改元素導(dǎo)致錯(cuò)誤的解決方法,針對(duì)增刪元素后出現(xiàn)dictionary changed size during iteration的異常解決做出討論和解決,需要的朋友可以參考下
2016-05-05
Python列表reverse()函數(shù)使用方法詳解
這篇文章主要詳細(xì)介紹了Python列表reverse()函數(shù)使用方法，文章通過代碼示例講解的非常詳細(xì)，對(duì)我們的學(xué)習(xí)或工作有一定的幫助,需要的朋友可以參考下
2023-07-07
解決Python3 被PHP程序調(diào)用執(zhí)行返回亂碼的問題
今天小編就為大家分享一篇解決Python3 被PHP程序調(diào)用執(zhí)行返回亂碼的問題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2019-02-02
Django shell調(diào)試models輸出的SQL語句方法
今天小編就為大家分享一篇Django shell調(diào)試models輸出的SQL語句方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
2019-08-08
Python引入多個(gè)模塊及包的概念過程解析
這篇文章主要介紹了Python引入多個(gè)模塊及包的概念過程解析,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2020-09-09
基于python pygame實(shí)現(xiàn)的兔子吃月餅小游戲
pygame是用來開發(fā)游戲的一套基于SDL的模板,它可以是python創(chuàng)建完全界面化的游戲和多媒體程序,而且它基本上可以在任何系統(tǒng)上運(yùn)行,這篇文章主要給大家介紹了基于python pygame實(shí)現(xiàn)的兔子吃月餅小游戲的相關(guān)資料,需要的朋友可以參考下
2021-09-09