Tensorflow訓(xùn)練模型默認(rèn)占滿所有GPU的解決方案

更新時間：2023年05月10日 09:00:56 作者：石頭兒啊

這篇文章主要介紹了Tensorflow訓(xùn)練模型默認(rèn)占滿所有GPU的解決方案，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教

Tensorflow訓(xùn)練模型默認(rèn)占滿所有GPU問題

在使用gpu服務(wù)器訓(xùn)練tensorflow模型時，總是占滿顯存！

TensorFlow默認(rèn)的是占用所有GPU

因此我們需要手動設(shè)置使用的GPU編號以及單個GPU顯存占用比例

1.第一步需要在代碼中開頭加入

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # 按照PCI_BUS_ID順序從0開始排列GPU設(shè)備
os.environ["CUDA_VISIBLE_DEVICES"]=‘0' ?# 使用0號gpu（想使用其他編號GPU，對應(yīng)修改引號中的內(nèi)容即可）
os.environ["CUDA_VISIBLE_DEVICES"]=‘0,1' # 使用0號GPU和1號GPU

2.第二步需要將代碼中的sess = tf.Session()改為

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333) # 通過改變0.333可以改變占用顯存比例
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

per_process_gpu_memory_fraction=0.333代表的含義就是每個GPU進(jìn)程中使用顯存的上限為該GPU總量的1/3

3.如果想要在程序運(yùn)行過程中連續(xù)查看GPU信息

可以在終端使用該指令（執(zhí)行指令：watch -n 3 -d nvidia-smi # 每隔三秒輸出一次）（前提是設(shè)備中有合適的NVIDIA驅(qū)動）

解決tensorflow2.2把GPU顯存占滿

安裝了tensorflow-gpu后，運(yùn)行程序默認(rèn)是把GPU的內(nèi)存全部占滿的，有時我們不想全部占滿，可以這樣操作。

解決代碼

import tensorflow as tf 
import os
os.environ['CUDA_VISIBLE_DEVICES']="0" # 指定哪塊GPU訓(xùn)練
config=tf.compat.v1.ConfigProto() 
# 設(shè)置最大占有GPU不超過顯存的80%（可選）
# config.gpu_options.per_process_gpu_memory_fraction=0.8
config.gpu_options.allow_growth = True  # 設(shè)置動態(tài)分配GPU內(nèi)存
sess=tf.compat.v1.Session(config=config)

如圖：

總結(jié)

以上為個人經(jīng)驗(yàn)，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

Python控制臺輸出俄羅斯方塊移動和旋轉(zhuǎn)功能
這篇文章主要介紹了Python控制臺輸出俄羅斯方塊移動和旋轉(zhuǎn)功能,本文通過實(shí)例代碼給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下
2021-04-04
使用Python控制攝像頭拍照并發(fā)郵件
這篇文章主要介紹了使用Python控制攝像頭拍照并發(fā)郵件的相關(guān)知識，非常不錯，具有一定的參考借鑒價值,需要的朋友可以參考下
2019-04-04
Python使用unittest進(jìn)行有效測試的示例詳解
這篇文章主要介紹了如何使用?unittest?來編寫和運(yùn)行單元測試，希望通過閱讀本文，大家能了解?unittest?的基本使用方法，以及如何使用?unittest?中的斷言方法和測試用例組織結(jié)構(gòu)
2023-06-06
使用Python手工計算x的算數(shù)平方根,來自中國古人的數(shù)學(xué)智慧
本篇采用的計算方法既非二分法也非牛頓迭代法，而是把中國古代的手工計算平方根的方法轉(zhuǎn)成代碼來完成。代碼有點(diǎn)煩雜，算是拋磚引玉吧，期待高手們寫出更好的代碼來
2021-09-09
Django實(shí)現(xiàn)上傳圖片功能
這篇文章為大家詳細(xì)主要介紹了Django實(shí)現(xiàn)上傳圖片，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2021-08-08
Django時區(qū)詳解
這篇文章主要介紹了Django時區(qū)詳解，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2019-07-07
基于Tensorflow讀取MNIST數(shù)據(jù)集時網(wǎng)絡(luò)超時的解決方式
這篇文章主要介紹了基于Tensorflow讀取MNIST數(shù)據(jù)集時網(wǎng)絡(luò)超時的解決方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-06-06
django使用django-apscheduler 實(shí)現(xiàn)定時任務(wù)的例子
今天小編就為大家分享一篇django使用django-apscheduler 實(shí)現(xiàn)定時任務(wù)的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-07-07
Python交換字典鍵值對的四種方法實(shí)例
字典中有成對出現(xiàn)的鍵和值,但是字典中的鍵值對不是都能修改的,只有值才能修改,下面這篇文章主要給大家介紹了關(guān)于Python交換字典鍵值對的四種方法,需要的朋友可以參考下
2022-12-12
python數(shù)字圖像處理圖像的繪制詳解
這篇文章主要為大家介紹了python數(shù)字圖像處理圖像的繪制示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪
2022-06-06