Python查看windows下GPU的使用情况

目录

一、使用nvidia-smi查看Windows的CUDA版本及GPU信息

二、使用pynvml查看GPU使用情况的命令

三、python 中使用GPUti实时查看GPU状况

四、使用gpustat库实时监测GPU使用情况（Linux下可以，Windows下不行的，衍生问题暂时没有一个好的解决方案）

一、使用nvidia-smi查看Windows的CUDA版本及GPU信息

在cmd中输入如下命令：

nvidia-smi

二、使用pynvml查看GPU使用情况的命令

 首先安装nvidia-ml-py包：

代码及对应解释如下：

import pynvml
pynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(0) # 指定显卡号
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.total/1024**2) #总的显存大小（float）
print(meminfo.used/1024**2)  #已用显存大小（float）
print(meminfo.free/1024**2)  #剩余显存大小（float）
print(pynvml.nvmlDeviceGetCount())#显示有几块GPU

三、python 中使用GPUti实时查看GPU状况

首先pip安装 gputi包：

运行如下命令：

import GPUtil
GPUtil.showUtilization()

四、使用gpustat库实时监测GPU使用情况（Linux下可以，Windows下不行的，衍生问题暂时没有一个好的解决方案）

Linux下：

Windows下（失败）： 

首先安装gpustat包：

在cmd中输入如下命令：

gpustat --w

报错：

ModuleNotFoundError: No module named '_curses'

其实好像是curses库不支持Windows。

这个问题可以解决，先使用where python命令找到自己python的安装路径：

 再运行如下代码查看自己python的版本：

import platform
print(platform.python_version())

在如下链接下载对应版本的curses包：

https://www.lfd.uci.edu/~gohlke/pythonlibs/#curses

下载你自己Python对应的版本！cp39表示py3.9   64表示64位

把whl文件放入Script文件夹后，进入Script文件夹：

重新在cmd中输入：

gpustat --w

又报错：

ModuleNotFoundError: No module named 'fcntl'

这个报错也可以解决：

在 python 安装目录 中 Lib目录（ 比如：D:Python39Lib ），创建 fcntl.py ，内容如下:

def fcntl(fd, op, arg=0):
    return 0

def ioctl(fd, op, arg=0, mutable_flag=True):
    if mutable_flag:
        return 0
    else:
        return ""

def flock(fd, op):
    return

def lockf(fd, operation, length=0, start=0, whence=0):
    return

当然到这里你就没问题了是最好的，如果还报错：

 ModuleNotFoundError: No module named ‘termios’

那你就凉凉了~~

这个东西是linux下的 windows没有，所以你不用查了，我翻了好多博客都没有解决方案，唯一的一个解答是（变通方法就超级麻烦了，这里仅提供一个学习链接）：

Airflow | 脚本东零西散？Airflow 快速搭建 pipeline（超详细）_HinGwenWoong的博客-CSDN博客