Keras設(shè)定GPU使用內(nèi)存大小方式(Tensorflow backend)
更新時(shí)間：2020年05月22日 17:14:10 作者：mjiansun
這篇文章主要介紹了Keras設(shè)定GPU使用內(nèi)存大小方式(Tensorflow backend)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
通過(guò)設(shè)置Keras的Tensorflow后端的全局變量達(dá)到。
import os
import tensorflow as tf
import keras.backend.tensorflow_backend as KTF
def get_session(gpu_fraction=0.3):
'''Assume that you have 6GB of GPU memory and want to allocate ~2GB'''
num_threads = os.environ.get('OMP_NUM_THREADS')
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=gpu_fraction)
if num_threads:
return tf.Session(config=tf.ConfigProto(
gpu_options=gpu_options, intra_op_parallelism_threads=num_threads))
else:
return tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
使用過(guò)程中顯示的設(shè)置session:
import keras.backend.tensorflow_backend as KTF
KTF.set_session(get_session())
補(bǔ)充知識(shí)：限制tensorflow的運(yùn)行內(nèi)存 （keras.backend.tensorflow）
我就廢話不多說(shuō)了，大家還是直接看代碼吧！
import tensorflow as tf from keras.backend.tensorflow_backend import set_session config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.5 #half of the memory set_session(tf.Session(config=config))
以上這篇Keras設(shè)定GPU使用內(nèi)存大小方式(Tensorflow backend)就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
解決python3 Pycharm上連接數(shù)據(jù)庫(kù)時(shí)報(bào)錯(cuò)的問(wèn)題
今天小編就為大家分享一篇解決python3 Pycharm上連接數(shù)據(jù)庫(kù)時(shí)報(bào)錯(cuò)的問(wèn)題，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-12-12
這篇文章主要介紹了python中正則表達(dá)式的使用詳解,另附上一些常用的Python正則例子，有需要的朋友可以參考下2014-10-10
今天小編就為大家分享一篇Python 將Matrix、Dict保存到文件的方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2018-10-10
詳解tensorflow訓(xùn)練自己的數(shù)據(jù)集實(shí)現(xiàn)CNN圖像分類
本篇文章了tensorflow訓(xùn)練自己的數(shù)據(jù)集實(shí)現(xiàn)CNN圖像分類，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2018-02-02
Python中ModuleNotFoundError錯(cuò)誤的問(wèn)題解決
本文主要介紹了Python中ModuleNotFoundError錯(cuò)誤的問(wèn)題解決，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2023-02-02
在Python編程語(yǔ)言中，變量是用于存儲(chǔ)數(shù)據(jù)值的標(biāo)識(shí)符，它們可以用來(lái)引用數(shù)據(jù)值，而不是直接使用值本身，本文將詳細(xì)介紹python中的變量，感興趣的同學(xué)可以參考閱讀2023-05-05
python書(shū)籍信息爬蟲(chóng)實(shí)例
這篇文章主要為大家詳細(xì)介紹了python書(shū)籍信息爬蟲(chóng)示例，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-03-03
解決Django生產(chǎn)環(huán)境無(wú)法加載靜態(tài)文件問(wèn)題的解決
這篇文章主要介紹了解決Django生產(chǎn)環(huán)境無(wú)法加載靜態(tài)文件問(wèn)題的解決，小編覺(jué)得挺不錯(cuò)的，現(xiàn)在分享給大家，也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2019-04-04
python中強(qiáng)大的format函數(shù)實(shí)例詳解
python中format函數(shù)用于字符串的格式化，這篇文章主要介紹了python中強(qiáng)大的format函數(shù),需要的朋友可以參考下2018-12-12
Python UI自動(dòng)化測(cè)試Web frame及多窗口切換
這篇文章主要為大家介紹了Python UI自動(dòng)化測(cè)試Web frame及多窗口切換，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-11-11