python如何使用gpu运算,普通python能不能调用GPU

　　需要使用pynvml库

　　现阶段点安装的命令为：

　　安装英伟达-ml-py

　　具体实例

　　导入pynvml

　　pynvml.nvmlInit()

　　# 这里的一是GPU id

　　handle=py nvml。nvmldevicegethandlebyindex(1)

　　meminfo=py nvml。nvmldevicegetmemoryinfo(句柄)

　　print(meminfo.total) #第二块显卡总的显存大小

　　print(meminfo.used)#这里是字节字节，所以要想得到以兆M为单位就需要除以1024**2

　　print(meminfo.free) #第二块显卡剩余显存大小

　　输出是：

　　11721506816

　　5333057536

　　6388449280

　　经过计算，总的现存大小11721506816/1024 /1024=11178M，已使用5333057536/1024 /1024=5086M，然后在人的本质终端中输入英伟达-smi查看显卡信息：

　　其他实例

　　从pynvml导入*

　　nvmlInit()

　　打印"驱动程序版本："，nvmlSystemGetDriverVersion()#显卡驱动版本

　　驱动程序版本：304.00

　　设备计数=nvmlDeviceGetCount()#几块显卡

　　对于范围内的我(设备计数):

　　.handle=nvmlDeviceGetHandleByIndex(I)

　　.打印"设备"，我，:，nvmlDeviceGetName(handle) #具体是什么显卡

　　.

　　设备0:特斯拉C2070

　　nvmlShutdown()

郑重声明：本文由网友发布，不代表盛行IT的观点，版权归原作者所有，仅为传播更多信息之目的，如有侵权请联系，我们将第一时间修改或删除，多谢。