python 输出 GPU内存 最大使用率

  • 安装 pip install nvidia-ml-py
from pynvml import *
def nvidia_info():# pip install nvidia-ml-pynvidia_dict = {"state": True,"nvidia_version": "","nvidia_count": 0,"gpus": []}try:nvmlInit()nvidia_dict["nvidia_version"] = nvmlSystemGetDriverVersion()nvidia_dict["nvidia_count"] = nvmlDeviceGetCount()for i in range(nvidia_dict["nvidia_count"]):handle = nvmlDeviceGetHandleByIndex(i)memory_info = nvmlDeviceGetMemoryInfo(handle)gpu = {"gpu_name": nvmlDeviceGetName(handle),"total": memory_info.total,"free": memory_info.free,"used": memory_info.used,"temperature": f"{nvmlDeviceGetTemperature(handle, 0)}℃","powerStatus": nvmlDeviceGetPowerState(handle)}nvidia_dict['gpus'].append(gpu)except NVMLError as _:nvidia_dict["state"] = Falseexcept Exception as _:nvidia_dict["state"] = Falsefinally:try:nvmlShutdown()except:passreturn nvidia_dictdef check_gpu_mem_usedRate():max_rate = 0.0while True:info = nvidia_info()# print(info)used = info['gpus'][0]['used']tot = info['gpus'][0]['total']print(f"GPU0 used: {used}, tot: {tot}, 使用率:{used/tot}")if used/tot > max_rate:max_rate = used/totprint("GPU0 最大使用率:", max_rate)

在跑任务时,另外运行脚本调用 check_gpu_mem_usedRate 就可以知道最大的 GPU内存 使用率,线上服务不要用的太满,最大80%左右为宜,防止极端情况GPU显存溢出

参考:
python获取GPU,CPU,硬盘,内存,系统,用户使用情况信息
【Python管理GPU】pynvml工具的安装与使用

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/471214.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

oracle 查看服务器密码修改,如何修改oracle用户密码

修改oracle用户密码的方法:首先连接oracle数据库所在服务器,并进入oracle控制台;然后输入“select username from dba_users”查看用户列表;最后输入修改用户口令即可。本教程操作环境:windows7系统、oracle版&#xf…

计算机组成原理——校验码(奇偶校验码、汉明校验码、循环冗余校验码)

一、为什么要使用校验码? 数据在计算机系统内加工、存取和传送的过程中可能会产生错误。为了减少和避免这类错误,引入了数据校验码。数据校验码是一种常用的带有发现某些错误,甚至带有一定自动改错能力的数据编码方法。 例子: …

(转)基因芯片数据GO和KEGG功能分析

随着人类基因组计划(Human Genome Project)即全部核苷酸测序的即将完成,人类基因组研究的重心逐渐进入后基因组时代(Postgenome Era),向基因的功能及基因的多样性倾斜。通过对个体在不同生长发育阶段或不同生理状态下大量基因表达的平行分析,…

LeetCode 2164. 对奇偶下标分别排序

文章目录1. 题目2. 解题1. 题目 给你一个下标从 0 开始的整数数组 nums 。根据下述规则重排 nums 中的值: 按 非递增 顺序排列 nums 奇数下标 上的所有值。 举个例子,如果排序前 nums [4,1,2,3] ,对奇数下标的值排序后变为 [4,3,2,1] 。奇…

九、操作系统——基本地址变换机构(详解)

一、概览 重点理解、记忆基本地址变换机构(用于实现逻辑地址到物理地址转换的一组硬件机构)的原理和流程 二、基本地址变换机构 基本地址变换机构可以借助进程的页表将逻辑地址转换为物理地址。 通常会在系统中设置一个页表寄存器(PTR&am…

客户端显示服务器图片不显示,客户端请求服务器图片不显示

客户端请求服务器图片不显示 内容精选换一换已成功登录鲲鹏代码迁移工具。只有管理员用户(portadmin)可以执行生成CSR文件、导入web服务器证书、重启和更换工作密钥的操作。普通用户只能查看web服务端证书信息。SSL证书通过在客户端浏览器和web服务器之间建立一条SSL安全通道(访…

LeetCode 2165. 重排数字的最小值(计数)

文章目录1. 题目2. 解题1. 题目 给你一个整数 num 。重排 num 中的各位数字,使其值 最小化 且不含 任何 前导零。 返回不含前导零且值最小的重排数字。 注意,重排各位数字后,num 的符号不会改变。 示例 1: 输入:nu…

2的次方表(1~64次方)

2的次方表(1~64次方) 2的1次方:2 2的2次方:4 2的3次方:8 2的4次方:16 2的5次方:32 2的6次方:64 2的7次方:128 2的8次方:256 2的9次方:512 2的10次…

Little Alchemy 游戏 - 扒答案。。 Java

Little Alchemy 是个用四个基本元素组合生成的小游戏。。。连Excalibur都可以有 写的代码是从这个游戏的hint页面得到答案的 1 import java.io.*;2 import java.net.URL;3 import java.util.*;4 5 public class Main {6 7 public static void main(String[] args) {8 …

优酷路由宝无线服务器,优酷路由宝一站式刷潘多拉!!!整理版

本帖最后由 louis000 于 2015-11-3 14:57 编辑优酷路由宝L1W刷潘多拉固件-整理优化版准备工作:使用有线连接路由宝和电脑;电脑网络连接设置为手动获取IP地址。---------------------------------------------------------------------------------------…

LeetCode 2166. 设计位集(Bitset)

文章目录1. 题目2. 解题1. 题目 位集 Bitset 是一种能以紧凑形式存储位的数据结构。 请你实现 Bitset 类。 Bitset(int size) 用 size 个位初始化 Bitset ,所有位都是 0 。void fix(int idx) 将下标为 idx 的位上的值更新为 1 。如果值已经是 1 ,则不…

Web框架——Flask系列之Flask简介(一)

一、Web应用程序作用 Web(World Wide Web)诞生最初的目的,是为了利用互联网交流工作文档 二、关于Web框架 (一)什么是Web框架? 已经封装好了一段代码,协助程序快速开发,相当于项目半成品开发者只需要按照框架约定要求,在指定位置写上自己…

服务器系统网卡驱动装不上,网卡驱动装不上去怎么办?

满意答案hgtn1q6xvv52014.09.28采纳率:58% 等级:13已帮助:9534人现提示原: SP2更考虑安全问题IP设置自获取其实DHCP服务器获取IP及相关参数程能由于某种原没完. SP1代种情况操作系统防止脱网(微软写)自作主张网卡设置169.x.x.x保留IP作任何…

moco-globalsettings

1.单个json文件 (1)目录配置 (2)新建foo.json 在project文件夹中新建foo.json [{"request" : {"uri" : "/foo"},"response" : {"text" : "foo"}} ] (3&a…

十三、MySQL存储过程相关知识总结 + 案例讲解(强化)

一、存储过程VS函数 存储过程和函数:类似于java中的方法 好处: 1、提高代码的重用性 2、简化操作 二、存储过程 含义:一组预先编译好的SQL语句的集合,理解成批处理语句 1、提高代码的重用性 2、简化操作 3、减少了编译次数并且…

LeetCode 2169. 得到 0 的操作数

文章目录1. 题目2. 解题1. 题目 给你两个 非负 整数 num1 和 num2 。 每一步 操作 中,如果 num1 > num2 ,你必须用 num1 减 num2 ; 否则,你必须用 num2 减 num1 。 例如,num1 5 且 num2 4 ,应该用 …

阿里云 腾讯云 服务器挂载数据盘

在默认拥有的腾讯云服务器是没有额外的数据盘的,默认Linux只有8GB系统盘,一般的网站也足够使用,如果额外购买的数据盘安装系统之后根据不同的面板、系统的路径问题可能不会自动加载到指定的数据盘目录,需要我们手工进行加载数据盘…

LeetCode 2171. 拿出最少数目的魔法豆(排序)

文章目录1. 题目2. 解题1. 题目 给你一个 正 整数数组 beans ,其中每个整数表示一个袋子里装的魔法豆的数目。 请你从每个袋子中 拿出 一些豆子(也可以 不拿出),使得剩下的 非空 袋子中(即 至少 还有 一颗 魔法豆的袋…

十四、MySQL函数相关知识总结(简单易懂)

函数 含义:一组预先编译好的SQL语句的集合,理解成批处理语句 提高代码的重用性简化操作减少了编译次数并且减少了和数据库服务器的连接次数,提高了效率 区别: 存储过程:可以有0个返回,也可以有多个返回…

linux tar order

http://blog.csdn.net/imyang2007/article/details/7634470 把常用的tar解压命令总结下,当作备忘: tar -c: 建立压缩档案-x:解压-t:查看内容-r:向压缩归档文件末尾追加文件-u:更新原压缩包中的文件 这五个是…