【Python管理GPU】pynvml工具的安装与使用
可以利用python工具pynvml来实现显卡信息的读取与管理
Nvidia的显卡提供了NVML(英伟达显卡管理库)以及构建在其上的nvidia-smi(显卡系统管理界面),可以方便的查询显卡的信息和工作状况。在python中同样可以利用pynvml库来实现显卡信息的获取。
1.安装
可以使用pip
方便的安装:
pip install nvidia-ml-py#也可以根据python版本制定2/3
#python2
pip install nvidia-ml-py2#python3
pip install nvidia-ml-py3#或者使用源码安装
#下载链接:http://pypi.python.org/pypi/nvidia-ml-py/
sudo python setup.py install#e.g
~$ pip install nvidia-ml-py3
>>>
Collecting nvidia-ml-py3
>>> Downloading
>>> https://files.pythonhosted.org/packages/6d/64/cce82bddb80c0b0f5c703bbdafa94bfb69a1c5ad7a79cff00b482468f0d3/nvidia-ml-py3-7.352.0.tar.gz
>>> Building wheels for collected packages: nvidia-ml-py3
>>> Running setup.py bdist_wheel for nvidia-ml-py3 ... done
>>> Stored in directory: xxxxxxx/xxxxxx/xxxxx
Successfully built nvidia-ml-py3
Installing collected packages: nvidia-ml-py3
Successfully installed nvidia-ml-py3-7.352.0
2.使用
#简单使用
from pynvml import *
nvmlInit() #初始化
print("Driver: "nvmlSystemGetDriverVersion()) #显示驱动信息
#>>> Driver: 384.xxx#查看设备
deviceCount = nvmlDeviceGetCount()
for i in range(deviceCount):handle = nvmlDeviceGetHandleByIndex(i)print("GPU", i, ":", nvmlDeviceGetName(handle))
#>>>
#GPU 0 : b'GeForce GTX 1080 Ti'
#GPU 1 : b'GeForce GTX 1080 Ti'#查看显存、温度、风扇、电源
handle = nvmlDeviceGetHandleByIndex(0)
info = nvmlDeviceGetMemoryInfo(handle)
print("Memory Total: ",info.total)
print("Memory Free: ",info.free)
print("Memory Used: ",info.used)print("Temperature is %d C"%nvmlDeviceGetTemperature(handle,0))
print("Fan speed is "nvmlDeviceGetFanSpeed(handle))
print("Power ststus",nvmlDeviceGetPowerState(handle))#最后要关闭管理工具
nvmlShutdown()#nvmlDeviceXXX有一系列函数可以调用,包括了NVML的大多数函数。
#具体可以参考:https://docs.nvidia.com/deploy/nvml-api/group__nvmlDeviceQueries.html#group__nvmlDeviceQueries
nvml:https://docs.nvidia.com/deploy/nvml-api/group__nvmlDeviceQueries.html#group__nvmlDeviceQueries
pypi:https://pypi.org/project/nvidia-ml-py/
usage:https://pythonhosted.org/nvidia-ml-py/
nvidia_smi:https://github.com/ultrabug/py3status/blob/master/py3status/modules/nvidia_smi.py
py3status:https://py3status.readthedocs.io/en/latest/modules.html
unsplash:https://unsplash.com/
pexels:https://www.pexels.com/
【Python管理GPU】pynvml工具的安装与使用相关推荐
- python 网页视频下载工具 you-get 安装及使用
公司网站上发布了视频都是YouTube上的,所以想要用一个方法把网站上的视频下载到本地,然后在网上找到了python3的web视频下载工具you-get. 首先需要安装python3,在python的 ...
- Python 之 Qt Designer工具相关安装配置
一.pyqt5.pyqt5-tool.paramiko.pyInstaller安装 直接在线安装: pip3 install PyQt5 pip3 install PyQt5-tools pip3 i ...
- python 输出 GPU内存 最大使用率
安装 pip install nvidia-ml-py from pynvml import * def nvidia_info():# pip install nvidia-ml-pynvidia_ ...
- Python学习笔记一简介及安装配置
Python学习笔记一 参考教程:廖雪峰官网https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e54 ...
- python包管理工具 ports_MacPorts安装使用Python/pip
Mac OS未预装任何在Unix/Linux中常见的命令行包管理工具,Mac OS中的App Store和自身的软件升级功能可以下载更新许多比较好的应用,但这些应用多数是满足普通消费者需求,对于开发人 ...
- linux下python安装包_Linux服务器中安装python包管理工具pip
pip是python的包管理工具,python的强大之处除了在于语法的简练,还有就是对众多的库支持了. 1.下载pip包管理工具 链接地址:https://pypi.python.org/pypi/p ...
- ubuntu安装nvidia显卡驱动+cuda9.0+cudnn7.0+查看cuda版本+安装tensorrt+python查看gpu显存
一,驱动安装 显卡驱动和cuda版本关系 卸载原先驱动 sudo apt-get remove --purge nvidia-\* ubuntu-drivers devices 查看显卡类型 Nvi ...
- python管理包管理工具pip和conda使用,及使用pip和conda创建虚拟环境
python管理包管理工具pip使用,及使用pip创建虚拟环境 文章目录: 1 pip的使用 1.1 pip的基础使用 1.1.1 pip安装库包(pip install) 1.1.2 pip卸载库包 ...
- Py之pipenv:Python包的管理利器pipenv简介、安装、使用方法详细攻略
Py之pipenv:Python包的管理利器pipenv简介.安装.使用方法详细攻略 目录 pipenv简介 pipenv安装 pipenv使用方法 pipenv简介 Python开发者应该听过pip ...
最新文章
- VS2010/MFC编程入门之十七(对话框:文件对话框)
- outlook html阅读,Html Email 邮件html页编写指南
- LiveVideoStackCon 2022 上海站 专题抢先看(3)
- Cisco开CASE方法
- ABPZero系列教程之拼多多卖家工具
- 自定义地图怎么做成html,自定义html为谷歌地图制作标记
- 4.安卓基础之Activity跳转动画
- C语言循环设计结构,c语言---循环结构设计.ppt
- 利用python scapy包进行抓包发包与ARP扫描
- Memcached与Redis的区别和选择
- 解析淘宝商城缘何更名“天猫”
- 《Java并发编程的艺术》:第1章 并发编程的挑战
- 计算机cad模板样例,教你如何新建适合自己的CAD模板
- 程序员跳槽时,如何正确做好职业规划?
- 《第一桶金怎么赚——淘宝开店创业致富一册通》一一1.1 创业者需具备的素质...
- 4.外部协作单位的组织与管理
- ABAP WB01 BDC ”No batch input data for screen “ ”没有屏幕 的批输入数据“
- 深度学习基础:SVD奇异值分解及其意义【转】
- 寅辞旧岁,卯定常虹丨ASKO洗碗机“净”护新春团圆时刻
- 牛客刷题和本地调试结果不一样--解决思路
热门文章
- OpenGL:关于获取渲染结果的深度信息的问题(二)
- MATLAB调用USB摄像头实现过程
- C++算法一:交换算法
- js字符串(String)转多维数组(Array) - 代码篇
- python cgi库_《Python 数据库 GUI CGI编程》
- 阿里云asp主机 后台登录一直提示验证码错误_吐槽一下阿里云关于网站备案的问题,真让人恶心生厌!...
- dedecms织梦模板|绿色仿下载吧网站织梦dede模板源码
- matlab期权定价模型比较,期权定价模型与数值方法(Matlab+Jupyter Notebook)
- sessionStorage、localStorage存储api
- 打开链接跳转浏览器代码