TensorFlow 2.0之后動(dòng)態(tài)分配顯存方式
更新時(shí)間：2022年12月08日 09:28:04 作者：scluis
這篇文章主要介紹了TensorFlow 2.0之后動(dòng)態(tài)分配顯存方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教
TensorFlow 2.0之后動(dòng)態(tài)分配顯存
import tensorflow as tf config = tf.compat.v1.ConfigProto() config.gpu_options.allow_growth = True sess = tf.compat.v1.Session(config = config) tf.compat.v1.keras.backend.set_session(sess)
keras和tensorflow動(dòng)態(tài)分配顯存
tensoflow或者基于tensorflow的keras，好像默認(rèn)會(huì)直接將GPU中所有的顯存都讀過(guò)來(lái)，然后慢慢用。
這樣如果是服務(wù)器是的話，別人就沒(méi)法用了，這里轉(zhuǎn)載了動(dòng)態(tài)分配顯存的方法：
tensorflow下，加入代碼：
config = tf.ConfigProto() config.gpu_options.allow_growth = True ? #設(shè)置動(dòng)態(tài)分配顯存 session = tf.Session(config=config, ...)
或者：
config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.3 ? #只一次調(diào)用30%的顯存 session = tf.Session(config=config, ...)
對(duì)于keras，如果使用tensorflow作為后端，多加一句話就好了：
import os os.environ["CUDA_VISIBLE_DEVICES"] = "1" from keras.backend.tensorflow_backend import set_session config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.3 set_session(tf.Session(config=config)) # 此處不同
總結(jié)
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了在Python中使用列表生成式的教程,列表生成式是Python具有的重要特性,需要的朋友可以參考下2015-04-04
python?pytorch圖像識(shí)別基礎(chǔ)介紹
大家好，本篇文章主要講的是python?pytorch圖像識(shí)別基礎(chǔ)介紹，感興趣的同學(xué)趕快來(lái)看一看吧，對(duì)你有幫助的話記得收藏一下2022-02-02
range函數(shù)大多數(shù)時(shí)常出現(xiàn)在for循環(huán)中,在for循環(huán)中可做為索引使用,下面這篇文章主要給大家介紹了關(guān)于Python中range函數(shù)的基本用法,文中通過(guò)實(shí)例代碼介紹的非常詳細(xì),需要的朋友可以參考下2022-01-01
Pandas DataFrame轉(zhuǎn)換為字典的方法
實(shí)際開(kāi)發(fā)中我們可能會(huì)遇到一類(lèi)問(wèn)題，如何將Pandas DataFrame轉(zhuǎn)換為字典，本文就來(lái)介紹一下，感興趣的可以了解一下2021-05-05
這篇文章主要為大家想想介紹了Python中Flask數(shù)據(jù)庫(kù)的連接與使用，文中的示例代碼講解詳細(xì)，具有一定的借鑒價(jià)值，感興趣的小伙伴可以學(xué)習(xí)一下2023-02-02
Python基礎(chǔ)類(lèi)繼承重寫(xiě)實(shí)現(xiàn)原理解析
這篇文章主要介紹了Python基礎(chǔ)類(lèi)繼承重寫(xiě)實(shí)現(xiàn)原理解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-04-04
python導(dǎo)出chrome書(shū)簽到markdown文件的實(shí)例代碼
python導(dǎo)出chrome書(shū)簽到markdown文件，主要就是解析chrome的bookmarks文件，然后拼接成markdown格式的字符串，最后輸出到文件即可。下面給大家分享實(shí)例代碼，需要的朋友參考下2017-12-12
Python如何優(yōu)雅刪除字符列表空字符及None元素
這篇文章主要介紹了Python如何優(yōu)雅刪除字符列表空字符及None元素,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-06-06
控制Python浮點(diǎn)數(shù)輸出位數(shù)的操作方法
在python的輸出結(jié)果中，尤其是浮點(diǎn)數(shù)的輸出，當(dāng)我們需要寫(xiě)入文本文件時(shí)，最好是采用統(tǒng)一的輸出格式，這樣也能夠增強(qiáng)結(jié)果的可讀性，這篇文章主要介紹了控制Python浮點(diǎn)數(shù)輸出位數(shù)的方法,需要的朋友可以參考下2022-04-04
使用Python和scikit-learn創(chuàng)建混淆矩陣的示例詳解
這篇文章主要介紹了使用Python和scikit-learn創(chuàng)建混淆矩陣的示例詳解,該示例包括生成數(shù)據(jù)集、為數(shù)據(jù)集選擇合適的機(jī)器學(xué)習(xí)模型、構(gòu)建、配置和訓(xùn)練它，最后解釋結(jié)果，即混淆矩陣，需要的朋友可以參考下2022-06-06