def __queryGPUProcessStats(self): """ Query stats for all GPU processes. Returns: gpuProcessStats (list): GPU process statistics. """ processes = GPUtil.getGPUProcesses() if processes: return [ GPUProcessStat(proc.pid, proc.processName, proc.gpuId, proc.gpuUuid, proc.usedMemory) for proc in processes ] return []
# Print os and python version information print('OS: ' + sys.platform) print(sys.version) # Print package name and version number print(GPU.__name__ + ' ' + GPU.__version__) # Get all GPUs gpus = GPU.getGPUs() print("All gpus:") for gpu in gpus: print(gpu) print() # Get all GPU processes gpuProcesses = GPU.getGPUProcesses() print("All gpu processes:") for gpuProcess in gpuProcesses: print(gpuProcess) print() # Show the utilization of all GPUs in a nice table GPU.showUtilization() # Show all stats of all GPUs in a nice table GPU.showUtilization(all=True) # Get all available GPU(s), ordered by ID in ascending order print('All available ordered by id: '), print(GPU.getAvailable(order='first', limit=999))