前言:
而今小伙伴们对“python怎么用gpu运行代码”可能比较重视,大家都想要学习一些“python怎么用gpu运行代码”的相关文章。那么小编同时在网上网罗了一些有关“python怎么用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 pynvmlpynvml.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 GPUtilGPUtil.showUtilization()四、使用gpustat库实时监测GPU使用情况(Linux下可以,Windows下不行的,衍生问题暂时没有一个好的解决方案)Linux下:Windows下(失败):
首先安装gpustat包:
在cmd中输入如下命令:
gpustat --w
报错:
ModuleNotFoundError: No module named '_curses'其实好像是curses库不支持Windows。
这个问题可以解决,先使用where python命令找到自己python的安装路径:
再运行如下代码查看自己python的版本:
import platformprint(platform.python_version())
在如下链接下载对应版本的curses包:
下载你自己Python对应的版本! cp39表示py3.9 64表示64位
把whl文件放入Script文件夹后,进入Script文件夹:
重新在cmd中输入:
gpustat --w
又报错:
ModuleNotFoundError: No module named 'fcntl'
这个报错也可以解决:
在 python 安装目录 中 Lib目录( 比如:D:\Python39\Lib ),创建 fcntl.py ,内容如下:
def fcntl(fd, op, arg=0): return 0def ioctl(fd, op, arg=0, mutable_flag=True): if mutable_flag: return 0 else: return ""def flock(fd, op): returndef lockf(fd, operation, length=0, start=0, whence=0): return当然到这里你就没问题了是最好的,如果还报错:
ModuleNotFoundError: No module named ‘termios’
那你就凉凉了~~