python怎么让系统用gpu跑

2024年11月29日 阅读 (51)

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

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

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

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

在cmd中输入如下命令:

nvidia-smi
python怎么让系统用gpu跑(1)

首先安装nvidia-ml-py包:

python怎么让系统用gpu跑(2)

代码及对应解释如下:

importpynvmlpynvml.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怎么让系统用gpu跑(3)

首先pip安装 gputi包:

python怎么让系统用gpu跑(4)

运行如下命令:

importGPUtilGPUtil.showUtilization()
python怎么让系统用gpu跑(5)
python怎么让系统用gpu跑(6)

首先安装gpustat包:

python怎么让系统用gpu跑(7)

在cmd中输入如下命令:

gpustat--w
python怎么让系统用gpu跑(8)

报错:

ModuleNotFoundError:Nomodulenamed'_curses'
python怎么让系统用gpu跑(9)

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

python怎么让系统用gpu跑(10)

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

importplatformprint(platform.python_version())
python怎么让系统用gpu跑(11)

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

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

python怎么让系统用gpu跑(12)

把whl文件放入Script文件夹后,进入Script文件夹:

python怎么让系统用gpu跑(13)
python怎么让系统用gpu跑(14)
python怎么让系统用gpu跑(15)

重新在cmd中输入:

gpustat--w

又报错:

ModuleNotFoundError:Nomodulenamed'fcntl'
python怎么让系统用gpu跑(16)

这个报错也可以解决:

在 python 安装目录 中 Lib目录,创建 fcntl.py ,内容如下:

deffcntl(fd, op, arg=0):return0defioctl(fd, op, arg=0, mutable_flag=True):ifmutable_flag:return0else:return""defflock:returndeflockf(fd, operation, length=0, start=0, whence=0):return
python怎么让系统用gpu跑(17)
ModuleNotFoundError:Nomodulenamed‘termios’

那你就凉凉了~~

郑重声明:玄微运势的内容来自于对中国传统文化的解读,对于未来的预测仅供参考。