TensorFlow和keras中GPU使用的設(shè)置操作
1. 訓(xùn)練運(yùn)行時候指定GPU
運(yùn)行時候加一行代碼：
CUDA_VISIBLE_DEVICES=1 python train.py
2. 運(yùn)行過程中按需或者定量分配GPU
tensorflow直接在開啟Session時候加幾行代碼就行，而Keras指定GPU，并限制按需用量和TensorFlow不太一樣，因?yàn)閗eras訓(xùn)練是封裝好的，不好對Session操作。如下是兩種對應(yīng)的操作。
keras中的操作：
import os import tensorflow as tf from keras.backend.tensorflow_backend import set_session # 指定第一塊GPU可用 os.environ["CUDA_VISIBLE_DEVICES"] = "0" #指定GPU的第二種方法 config = tf.ConfigProto() config.gpu_options.allocator_type = 'BFC' #A "Best-fit with coalescing" algorithm, simplified from a version of dlmalloc. config.gpu_options.per_process_gpu_memory_fraction = 0.3 #定量 config.gpu_options.allow_growth = True #按需 set_session(tf.Session(config=config))
TensorFlow中的操作：
#指定GPU import os os.environ["CUDA_VISIBLE_DEVICES"] = "0" #設(shè)置GPU定量分配 config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的顯存 session = tf.Session(config=config) #設(shè)置GPU按需分配 config = tf.ConfigProto() config.gpu_options.allow_growth = True session = tf.Session(config=config)
補(bǔ)充：Keras以及Tensorflow強(qiáng)制使用CPU，GPU
Keras如果是使用Theano后端的話，應(yīng)該是自動不使用ＧＰＵ只是用ＣＰＵ的，啟動ＧＰＵ使用Theano內(nèi)部命令即可。
對于Tensorflow后端的Keras以及Tensorflow會自動使用可見的ＧＰＵ，而我需要其必須只運(yùn)行在ＣＰＵ上。網(wǎng)上查到三種方法，最后一種方法對我有用，但也對三種都做如下記錄：
使用tensorflow的 with tf.device('/cpu:0'):函數(shù)。簡單操作就是把所有命令都放在前面所述的域里面。
使用tensorflow聲明Session時的參數(shù)： 關(guān)于tensorflow中Session中的部分參數(shù)設(shè)置，以及Keras如何設(shè)置其調(diào)用的Tensorflow的Session，可以參見Keras設(shè)定GPU使用內(nèi)存大小(Tensorflow backend)。
對于Tensorflow，聲明Session的時候加入device_count={'gpu':0}即可，代碼如下：
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(device_count={'gpu':0}))
對于Keras,則調(diào)用后端函數(shù)，設(shè)置其使用如上定義的Session即可，代碼如下：
import tensorflow as tf
import keras.backend.tensorflow_backend as KTF
KTF.set_session(tf.Session(config=tf.ConfigProto(device_count={'gpu':0})))
對于多線程以及ＧＰＵ內(nèi)存設(shè)置等可以參見Keras設(shè)定GPU使用內(nèi)存大小(Tensorflow backend)。
3、第三種是使用CUDA_VISIBLE_DEVICES命令行參數(shù)，代碼如下：
CUDA_VISIBLE_DEVICES="0" python3 train.py
以上為個人經(jīng)驗(yàn)，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python?SDK實(shí)現(xiàn)私服上傳下載的示例
本文主要介紹了Python?SDK實(shí)現(xiàn)私服上傳下載的示例，文中通過示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下<BR>2021-11-11
在Python編程中，有一種強(qiáng)大的技術(shù)允許我們在運(yùn)行時修改或生成代碼，這就是元編程，Python的元編程工具包括裝飾器，元類，以及函數(shù)和類的各種動態(tài)修改技術(shù)，這篇文章將向你介紹元編程的基本概念，并通過實(shí)例講解如何使用元編程2023-07-07
使用python+pandas讀寫xlsx格式中的數(shù)據(jù)
這篇文章主要介紹了使用python+pandas讀寫xlsx格式中的數(shù)據(jù)，文章圍繞主題展開詳細(xì)的內(nèi)容介紹，具有一定的參考價值，感興趣的小伙伴可以參考一下2022-08-08
今天小編就為大家分享一篇pandas 使用均值填充缺失值列的小技巧分享，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07
這篇文章主要介紹了pyftplib中文亂碼問題解決方案,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-01-01
這篇文章主要為大家分享了如何利用Python中的matplotlib庫實(shí)現(xiàn)繪制箱線圖與異常值的輸出，文中的示例代碼講解詳細(xì)，需要的可以參考一下2022-04-04
這篇文章主要介紹了python判斷字符串的前兩個字母是否是”id",使用 Python 的字符串切片來判斷一個字符串的前兩個字母是否是 "id"，本文結(jié)合示例代碼給大家介紹的非常詳細(xì)，需要的朋友可以參考下2023-04-04
Python并發(fā)編程的幾種實(shí)現(xiàn)方式
Python并發(fā)編程讓程序能同時執(zhí)行多個任務(wù),Python提供多種并發(fā)實(shí)現(xiàn)方式,包括多線程、多進(jìn)程、異步編程等,多線程受全局解釋器鎖（GIL）影響,在CPU密集型任務(wù)上不提升性能,但適合I/O密集型任務(wù),多進(jìn)程適用于CPU密集型任務(wù)2024-10-10
python中關(guān)于對super()函數(shù)疑問解惑
Python中的super()是用于調(diào)用父類(或父類的父類...)方法的函數(shù),主要用于多繼承,單繼承問題不大,下面這篇文章主要給大家介紹了關(guān)于python中關(guān)于對super()函數(shù)疑問解惑的相關(guān)資料,需要的朋友可以參考下2022-08-08
Python實(shí)現(xiàn)連接MySQL數(shù)據(jù)庫的常見方法總結(jié)
這篇文章主要為大家介紹了兩種Python中用來連接 MySQL 數(shù)據(jù)庫的方法，并且針對這兩種方法，我們還將對代碼進(jìn)行封裝和優(yōu)化，提高程序的可讀性和健壯性，需要的可以收藏一下2023-05-05