欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

Keras設(shè)定GPU使用內(nèi)存大小方式(Tensorflow backend)

 更新時(shí)間：2020年05月22日 17:14:10   作者：mjiansun  
這篇文章主要介紹了Keras設(shè)定GPU使用內(nèi)存大小方式(Tensorflow backend)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

通過設(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))

使用過程中顯示的設(shè)置session:

import keras.backend.tensorflow_backend as KTF
KTF.set_session(get_session())

補(bǔ)充知識(shí)：限制tensorflow的運(yùn)行內(nèi)存 （keras.backend.tensorflow）

我就廢話不多說了，大家還是直接看代碼吧！

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)就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

相關(guān)文章

  • python接口自動(dòng)化之正則用例參數(shù)化的示例詳解

    python接口自動(dòng)化之正則用例參數(shù)化的示例詳解

    這篇文章主要介紹了python接口自動(dòng)化之正則用例參數(shù)化,它是一個(gè)特殊的字符序列，它能幫助你方便的檢查一個(gè)字符串是否與某種模式匹配，本文給大家介紹的非常詳細(xì)，需要的朋友可以參考下
    2022-07-07
  • Python文件常見操作實(shí)例分析【讀寫、遍歷】

    Python文件常見操作實(shí)例分析【讀寫、遍歷】

    這篇文章主要介紹了Python文件常見操作,結(jié)合實(shí)例形式分析Python針對(duì)文件的讀寫、遍歷等相關(guān)操作技巧,需要的朋友可以參考下
    2018-12-12
  • 在Python中居然可以定義兩個(gè)同名通參數(shù)的函數(shù)

    在Python中居然可以定義兩個(gè)同名通參數(shù)的函數(shù)

    今天小編就為大家分享一篇在Python中居然可以定義兩個(gè)同名通參數(shù)的函數(shù)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
    2019-01-01
  • 使用python實(shí)現(xiàn)UDP通信方式

    使用python實(shí)現(xiàn)UDP通信方式

    這篇文章主要介紹了使用python實(shí)現(xiàn)UDP通信方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
    2022-05-05
  • python實(shí)現(xiàn)kmp算法的實(shí)例代碼

    python實(shí)現(xiàn)kmp算法的實(shí)例代碼

    這篇文章主要介紹了python實(shí)現(xiàn)kmp算法的實(shí)例代碼,非常不錯(cuò)，具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
    2019-04-04
  • 1 行 Python 代碼快速實(shí)現(xiàn) FTP 服務(wù)器

    1 行 Python 代碼快速實(shí)現(xiàn) FTP 服務(wù)器

    FTP 服務(wù)器，在此之前我都是使用Linux的vsftpd軟件包來搭建FTP服務(wù)器的，現(xiàn)在發(fā)現(xiàn)了利用pyftpdlib可以更加簡單的方法即可實(shí)現(xiàn)FTP服務(wù)器的功能。下面小編給大家?guī)砹? 行 Python 代碼快速實(shí)現(xiàn) FTP 服務(wù)器，需要的朋友參考下
    2018-01-01
  • 使用Python實(shí)現(xiàn)壓縮pptx的功能

    使用Python實(shí)現(xiàn)壓縮pptx的功能

    當(dāng)處理大型PPTX文件時(shí),其中包含許多高分辨率照片時(shí),文件大小可能會(huì)顯著增加,為了解決這個(gè)問題,我們可以使用Python編程語言和python-pptx庫來壓縮PPTX文件中的照片,下面我們就來看看具體操作吧
    2024-02-02
  • python 識(shí)別圖片中的文字信息方法

    python 識(shí)別圖片中的文字信息方法

    今天小編就為大家分享一篇python 識(shí)別圖片中的文字信息方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
    2018-05-05
  • python3+PyQt5自定義視圖詳解

    python3+PyQt5自定義視圖詳解

    這篇文章主要為大家詳細(xì)介紹了python3+PyQt5自定義視圖的相關(guān)資料，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
    2018-04-04
  • Python-split()函數(shù)實(shí)例用法講解

    Python-split()函數(shù)實(shí)例用法講解

    在本篇文章里小編給大家整理的是一篇關(guān)于Python-split()函數(shù)實(shí)例用法講解，有興趣的朋友們可以學(xué)習(xí)下。
    2020-12-12

最新評(píng)論