keras如何指定运行时显卡及限制GPU用量

更新时间：2024年03月28日 09:49:44 作者：T_白日梦想家

这篇文章主要介绍了keras如何指定运行时显卡及限制GPU用量问题,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

keras指定运行时显卡及限制GPU用量

keras在使用GPU的时候有个特点，就是默认全部占满显存。

这样如果有多个模型都需要使用GPU跑的话，那么限制是很大的，而且对于GPU也是一种浪费。

因此在使用keras时需要有意识的设置运行时使用那块显卡，需要使用多少容量。

这方面的设置一般有三种情况：

1. 指定显卡
2. 限制GPU用量
3. 即指定显卡又限制GPU用量

查看GPU使用情况语句（linux）

# 1秒钟刷新一次
watch -n 1 nvidia-smi

一、指定显卡

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "2"

这里指定了使用编号为2的GPU，大家可以根据需要和实际情况来指定使用的GPU

二、限制GPU用量

1、设置使用GPU的百分比

import tensorflow as tf
import keras.backend.tensorflow_backend as KTF

#进行配置，使用30%的GPU
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
session = tf.Session(config=config)

# 设置session
KTF.set_session(session )

需要注意的是，虽然代码或配置层面设置了对显存占用百分比阈值，但在实际运行中如果达到了这个阈值，程序有需要的话还是会突破这个阈值。

换而言之如果跑在一个大数据集上还是会用到更多的显存。

以上的显存限制仅仅为了在跑小数据集时避免对显存的浪费而已。

2、GPU按需使用

import tensorflow as tf
import keras.backend.tensorflow_backend as KTF

config = tf.ConfigProto()  
config.gpu_options.allow_growth=True   #不全部占满显存, 按需分配
session = tf.Session(config=config)

# 设置session
KTF.set_session(sess)

三、指定GPU并且限制GPU用量

这个比较简单，就是讲上面两种情况连上即可。。。

import os
import tensorflow as tf
import keras.backend.tensorflow_backend as KTF

# 指定第一块GPU可用 
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

config = tf.ConfigProto()  
config.gpu_options.allow_growth=True   #不全部占满显存, 按需分配
sess = tf.Session(config=config)

KTF.set_session(sess)

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

python 操作sqlite数据库的方法
sqlite的官网 sqlite.org/index.html SQLite 作为一个SQL数据库引擎，是由C语言实现的，又小又快，具有高可靠性且功能齐全。作为嵌入式数据库，在移动设备中使用非常广泛且方便。本文讲解如何使用python操作sqlite数据库
2021-06-06
django中的select_related和prefetch_related性能优化分析
这篇文章主要介绍了django中的select_related和prefetch_related性能优化分析,本文给大家介绍的非常详细,需要的朋友可以参考下
2024-07-07
Python实现推送百度链接的示例代码
有时为了提高搜索效率，也让搜索引擎更容易发现自己的文章，我们需要将文章链接推送到百度站长平台，起到快速收录的目的。本文将主要介绍如何通过Python实现这一功能，需要的可以参考一下
2021-12-12
Python下opencv库的安装过程及问题汇总
本文主要内容是python下opencv库的安装过程，涉及我在安装时遇到的问题，并且，将从网上搜集并试用的一些解决方案进行了简单的汇总，感兴趣的小伙伴们可以参考一下
2021-06-06
Python实现按中文排序的方法示例
这篇文章主要介绍了Python实现按中文排序的方法,涉及Python基于locale模块的中文编码与排序相关操作技巧,需要的朋友可以参考下
2018-04-04
Python psutil模块简单使用实例
这篇文章主要介绍了Python psutil模块简单使用实例,本文直接给出使用脚本,实现查看cpu的信息、查看内存信息、查看系统启动时间、查看网卡信息等,需要的朋友可以参考下
2015-04-04
python中Flask框架简单入门实例
这篇文章主要介绍了python中Flask框架简单入门实例,以一个hello程序简单分析了Flask框架的使用技巧,需要的朋友可以参考下
2015-03-03
Python进度条神器tqdm使用实例详解
Python进度条神器tqdm是一个快速、可扩展的进度条工具，可以轻松地为Python脚本添加进度条。它可以在循环中自动计算进度，并在终端中显示进度条，让用户了解程序的运行情况。tqdm还支持多线程和多进程，并且可以自定义进度条的样式和显示方式。
2023-06-06
Python实现获取命令行输出结果的方法
这篇文章主要介绍了Python实现获取命令行输出结果的方法,涉及Python命令执行及文件读写等相关操作技巧,需要的朋友可以参考下
2017-06-06
python机器学习XGBoost梯度提升决策树的高效且可扩展实现
这篇文章主要为大家介绍了python机器学习XGBoost梯度提升决策树的高效且可扩展实现,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2024-01-01