pip安裝pynvml模塊
1 import pynvml 2 pynvml.nvmlInit() 3 handle = pynvml.nvmlDeviceGetHandleByIndex(0) # 0表示第一塊顯卡 4 meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle) 5 print(meminfo.total/1024**2) #第二塊顯卡總的顯存大小 6 print(meminfo.used/1024**2) #這里是字節bytes，所以要想得到以兆M為單位就需要除以1024**2 7 print(meminfo.free/1024**2) #第二塊顯卡剩余顯存大小